使用yolonas 训练自己的数据集

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、终于开始训练了：

很开心啊。遇到一个又一个问题，不过都解决了。先是有些文件漏掉，没有改全。后来意识到把文件下下来修改是不行的。需要跑到安装目录去改相应文件。走了不少弯路。