火炬视觉更快的RCNN微调器
pytorch-fasterrcnn的Python项目详细描述
Pythorch更快RCNN
在Pytorch中更快地实现RCNN微调。在
如何使用?在
- git克隆回购
git clone https://github.com/oke-aditya/pytorch_fasterrcnn.git
- 为您的系统安装Pythorch和torchvision。在
只需编辑配置文件以设置超参数。在
- 按照csv文件进行验证和保存
注意
不要将target用作0类。保留为背景。在
^{pr2}$- 在
只需编辑配置文件以设置超参数
在 - 在
运行火车.py文件
在
特点:-
- 它适用于多类对象检测。在
支持的主干:-
- 在
请注意,主干是在imagenet上预先训练的。在
在 - 在
支持以下主干
在
- vgg11、vgg13、vgg16、vgg19
- resnet18、resnet34、resnet50、resnet101、resnet152
- renext101公司
- mobilenet_版本2
样本输出
头盔探测器
掩模探测器
如果你喜欢实现或者得到了灵感,一定要给一颗星星:-)
- 项目
标签: