【文件属性】:
文件名称:PyTorch-YOLOv3:YOLOv3的最小PyTorch实现
文件大小:2.69MB
文件格式:ZIP
更新时间:2021-02-18 03:54:41
Python
警告:此存储库已过时,因为我不幸没有时间对其进行维护。 如果您想作为合作者继续开发它,请给我发送电子邮件至 。
PyTorch-YOLOv3
YOLOv3的最小PyTorch实现,并支持训练,推断和评估。
安装
克隆和安装要求
$ git clone https://github.com/eriklindernoren/PyTorch-YOLOv3
$ cd PyTorch-YOLOv3/
$ sudo pip3 install -r requirements.txt
下载预先训练的体重
$ cd weights/
$ bash download_weights.sh
下载COCO
$ cd data/
$ bash get_coco_dataset.sh
测试
在COCO测试中评估模型。
$ python3 test.py --weights_path weights/yolov3
【文件预览】:
PyTorch-YOLOv3-master
----train.py(7KB)
----utils()
--------utils.py(12KB)
--------datasets.py(4KB)
--------parse_config.py(1KB)
--------__init__.py(0B)
--------transforms.py(3KB)
--------logger.py(790B)
--------augmentations.py(665B)
----test.py(4KB)
----detect.py(5KB)
----requirements.txt(85B)
----weights()
--------download_weights.sh(408B)
----models.py(14KB)
----config()
--------coco.data(115B)
--------create_custom_model.sh(8KB)
--------custom.data(99B)
--------yolov3.cfg(8KB)
--------yolov3-tiny.cfg(2KB)
----LICENSE(34KB)
----assets()
--------traffic.png(312KB)
--------messi.png(258KB)
--------dog.png(342KB)
--------giraffe.png(405KB)
----README.md(7KB)
----data()
--------custom()
--------samples()
--------coco.names(625B)
--------get_coco_dataset.sh(1KB)
----.gitignore(119B)