yolov5格式的香烟数据集
https://download.csdn.net/download/qq_42864343/88110620?spm=1001.2014.3001.5503
创建yolo-nas的运行环境
进入Pycharm的terminal,输入如下命令
conda create -n yolonas python=3.8
pip install super-gradients
使用自定义数据训练Yolo-nas
准备数据
在YOLO-NAS根目录下创建mydata文件夹(名字可以自定义),目录结构如下:
将自己数据集里用labelImg标注好的xml文件
放到xml目录
图片放到images目录
划分数据集
把划分数据集代码 split_train_val.py放到yolo-nas目录下: