说明:
在Ubuntu上崩了几次过后,终于Ubuntu坏了,出于各种原因目前暂时无法重装。本文档分为两个部分:识别和训练。
1、识别
参考链接:
/peacepeacepeace/p/9060229.html
/KID_yuan/article/details/88380269
先下载cuda:/cuda-downloads
配置cuda:/bmw601055/article/details/77073630
/qq_37296487/article/details/83028394
下了10.2打不开:
/p/94220564?utm_source=wechat_session
下老版本10.0:
/qq_36362060/article/details/78608580
或者更新显卡驱动:
/drivers
2、训练
参考链接:
/congcong7267/article/details/82981084
/taotingz/p/11319410.html
/clover_my/article/details/89925012
一、关于修改yolov3.cfg文件(这几个参数还不知道是什么含义,后面再找):
1、 将batch=1改为batch=64:
2、 将subdivisions改成8 :subdivisions=8
即上图batch下面那个
3、 修改classes和filters:
注意操作:先搜索yolo(共有三处),每处都要修改如图中的classes和filters。其中classes改为自己类的数量,filters改为filters=(classes+5)x3
其他地方的filters和classes不要改
4、 另外:如果你对显卡较差(4G显存以下),会出现内存溢出错误(Out of memory),改正方法:将batch改小些(64,32,16,8),将random改成0关闭多尺度训练。同样,random也是第3步搜索yolo后修改[yolo]下面的那个。
二、修改voc_label.py
使用代码在ImageSets\Main目录下生成test.txt(测试集)、train.txt(训练集)、val.txt(验证集)、trainval.txt(训练验证集,由train.txt和val.txt组成)。VOC中,test大概是整个数据集的50%,trainval是整个数据集剩下的50%;train大概是trainval的50%,val是trainval剩下的50%。所占比例可在代码中修改,以Python为例(其中trainval是整个数据集的70%)
报错:
/farces/blog/2209462 关闭多尺度输出或者改小batch,见上面修改yolov3.cfg
Yolo训练相关参数:
/weixin_42731241/article/details/81474920
训练再参考链接:
/p/af2d292da518
/tango1221/p/10084969.html
什么时候停止训练
/weixin_30847939/article/details/99257729?depth_1-utm_source=distribute.pc_relevant.none-task&utm_source=distribute.pc_relevant.none-task
yolo相关资料:
/anai/p/11465175.html