今日立冬
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改小了