资源介绍
PyTorch中的YOLOv3
YOLOv3在PyTorch中的实现非常少,仅跨越大约800行代码(不包括绘图功能等),支持训练和评估,并带有辅助功能以进行推理。 目前有Pascal-VOC的预训练权重,其中MS COCO即将发布。 只需在模型上对输出格式进行最小的更改,原始权重也可以无缝加载。
安装
克隆和安装要求
$ git clone https://github.com/SannaPersson/YOLOv3-PyTorch.git
$ cd YOLOv3-PyTorch
$ pip install requirements.txt
在Pascal-VOC上下载预训练重量
可以从以下页面下载Pascal-VOC的预训练砝码: ://www.kaggle.com/sannapersson/yolov3-weights-for-pascal-voc-with-781-map
下载原