YOLOV3的pytorch实现

版权声明：本文为博主原创文章，转载请联系作者 https://blog.csdn.net/u013832707/article/details/80083401

源代码

作者官方链接https://github.com/ayooshkathuria/pytorch-yolo-v3

运行

这个版本只实现了预测，并不能训练。同时运行十分简单，只需要你符合一定的条件。

直接运行

需要你满足一下条件

• Python3.5
• Opencv
• Pytorch0.3或者0.4
  如果你满足这些条件，并且在根目录下放进去yolov3.weight，执行命令python cam_demo.py就可运行

在python2.7下运行

博主的环境为Python2.7，所以就折腾了一会才成功运行。
修改源代码：
- 修改darknet.py地24行fwd = nn.Sequential(self.linear_1, *self.middle, self.output)，将self前的※号删除
- 修改darknet.py的249行mask = [int(x) for x in mask]将x改为其它的字母如num
- 将cam_demo.py中的119行im_dim = torch.FloatTensor(dim).repeat(1,2)取消注释
- 将cam_demo.py第149行colors = pkl.load(open(“pallete”, “rb”))改为colors = pkl.load(open(“pallete.pkl”, “rb”))，同时用博主给的文件替换源文件中的pallete文件

文件下载地址https://download.csdn.net/download/u013832707/10374075