調試YOLOv3/YOLOv5過程中遇到的問題匯總


今日立冬

yolov1

yolo1論文

https://www.bilibili.com/video/BV15w411Z7LG

yolo2論文

https://www.bilibili.com/video/BV1Q64y1s74K

yolo3論文

https://www.bilibili.com/video/BV1Vg411V7bJ

yolov3

yolo3視頻

https://www.bilibili.com/video/BV1Hp4y1y788

源碼地址

https://github.com/bubbliiiing/yolo3-pytorch

博客地址

https://blog.csdn.net/weixin_44791964/article/details/105310627

下載了好幾個版本的yolo3,只有這個博主的成功了。問題在視頻評論區,以及代碼中都有解決方法。

 

yolov5

yolo5視頻

https://www.bilibili.com/video/BV1tf4y1t7ru?p=1

官方項目地址

https://github.com/ultralytics/yolov5

標注數據集地址

https://www.makesense.ai/

c++生成工具

https://visualstudio.microsoft.com/zh-hans/visual-cpp-build-tools/

匯總了一下自己遇到的問題

bug1: pycococtools

pycococtools的安裝包https://pan.baidu.com/s/1nWQdPRtGwNnOO2DkxRuGuA提取碼:i5d7 安裝完直接解壓復制其中pycocotools兩個文件夾到conda環境…\Lib\site-packages之中

bug2: Cant get attribute SPPF on module models.common from xxx\\common.py

https://blog.csdn.net/RooKichenn/article/details/120866650

bug3:RuntimeError: Given groups=1, weight of size 【512, 1024, 1, 1】, expected input【1, 512, 8, 8】 to have 1024 channels, but got 512 channels instead

parser.add_argument('--cfg', type=str, default='models/yolov5s.yaml', help='model.yaml path')

bug4: coco128

運行train.py的時候,數據集自動下載之后,工程文件下會出現coco128.zip文件,報未解壓報錯,解決方法:coco128.zip 解壓到(yolov5-5.0文件夾同一目錄下文件夾叫coco128),因為coco128.yaml需要去train: ../coco128/images/train2017/ # 128 images
val: ../coco128/images/train2017/ # 128 images

bug5:BrokenPipeError: 【Errno 32】 Broken pipe

 根據文章https://blog.csdn.net/qq_33666011/article/details/81873217,我把workers改小了,暫時不報錯了。

 

bug6:RuntimeError: [enforce fail at …\c10\core\CPUAllocator.cpp:72] data.DefaultCPUAllocator: not enough memory: you tried to allocate xxx bytes.

私以為設置的張量過大,導致內存分配不足,在train.py中把batchsize改小了

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM