想多熟悉下pytorch,因而选择了yolov4-pytorch版本来熟悉整个流程。
代码下载地址:
https://github.com/Tianxiaomo/pytorch-YOLOv4
这位大神厉害啊,这么快就复现了,好好学习膜拜中!
yolov4.weights权重百度网盘下载地址:
https://pan.baidu.com/s/1dAGEW8cm-dqK14TbhhVetA 提取码:dm5b
下载后用pycharm打开工程。
个人更喜欢把参数写在代码中,所以将demo.py中main