Day11
尝试用yolov8s
和yolov5s
跑个baseline,脚本改了又改,300epochs跑出来过拟合了,跑一次200的试试
集群的内存太低了,workers只能设为2
参照b站视频改了一下yolov8,加了个CBAM模块,还是得看yolo结构,不然加模块的参数都不知道怎么设置
代码上传github了,感觉还是得把ultralytics的源码一块放进去,不然后面要修改网络模型还得去找虚拟环境的代码
顺便记一下虚拟环境里安装的库的路径~/.conda/envs/yolo/lib/python3.10/site-packages/ultralytics/nn
本文由作者按照 CC BY 4.0 进行授权