【yolov4】學習筆記一

想多熟悉下pytorch,因而選擇了yolov4-pytorch版本來熟悉整個流程。
代碼下載地址:
https://github.com/Tianxiaomo/pytorch-YOLOv4
這位大神厲害啊，這麼快就復現了,好好學習膜拜中！
yolov4.weights權重百度網盤下載地址:
https://pan.baidu.com/s/1dAGEW8cm-dqK14TbhhVetA 提取碼:dm5b

下載後用pycharm打開工程。

個人更喜歡把參數寫在代碼中，所以將demo.py中main部分改爲

if __name__ == '__main__':
    cfgfile = 'cfg/yolov4.cfg'
    weightfile = 'yolov4.weights'
    imgfile = 'data/dog.jpg'
    detect(cfgfile, weightfile, imgfile)

缺少哪個庫就安裝即可。

如果使用cuda,則修改demo.py中use_cuda = 1以及將utils/utils.py中第452行修改爲

boxes.append(get_region_boxes1(list_boxes[i].data.cpu().numpy(), 0.6, 80, masked_anchors, len(anchor_masks[i])))

先用cpu跑一下。
運行demo.py
cpu運行結果如下:(我這電腦是真的慢。。。)

gpu下面結果爲

predictions.jpg

後續會一步一步閱讀該代碼並分享心得。