1、首先安装super-gradients
2、数据集目录:
Images/train /val /test
labels/train /val /test
3、安装参考:
【YOLO_NAS训练自己的数据集】_yolo-nas 自定义数据集训练-CSDN博客
YOLO NAS 训练自己的数据集_yolonas训练自定义数据集-CSDN博客
4、之后开始训练数据。
使用命令运行:
python -m super_gradients.train_from_recipe --config-name=my_yolo_nas_s.yaml
5、出现错误:
UnicodeDecodeError: 'utf-8' codec can't decode byte 0xb7
6、之后遇到GPU 显存爆掉,把batch_size变小:
7、终于开始训练了:
很开心啊。遇到一个又一个问题,不过都解决了。先是有些文件漏掉,没有改全。后来意识到把文件下下来修改是不行的。需要跑到安装目录去改相应文件。走了不少弯路。