-
PyTorch版本的YOLOv3:对YOLOv3网络的PyTorch实现
资源介绍
在PyTorch中实现的YOLO v3对象检测器
该存储库包含YOLOv3的PyTorch实现论文。 为什么要执行此回购? 该代码以pythonic /“ pytorch ish ”样式实现了论文,并且将为PyTorch用户了解并使用YOLOv3对象检测器提供熟悉和(希望)帮助。 原始的预训练权重(在COCO数据集上)已转换为PyTorch的state_dict,可以在下载(不需要复杂的网络构建器和权重转换器,只需普通的torch.load() )。
要求
的Python 3
火炬(0.4)
火炬视觉
OpenCV
用法
回购包含2个可运行脚本。
检测:
$ python detect.py images/dog.jpg
如果要指定自己的砝码文件,请使用-weights path/to/weights或者如果要保存检测结果,请使用-save标志(注意:默认情况下,砝码应放置在da