基于detectron2的编译难度较大,本次教程抛开detectron2,完全依赖torch环境。
1、下载工程
https://github.com/laitathei/YOLOv7-Pytorch-Segmentation.git
2、安装依赖包
cd YOLOv7-Pytorch-Segmentation
pip install --upgrade pip
pip install -r requirements.txt
pip install labelme
3、准备数据
- Label you dataset image
labelme
4、转换json格式为yolo的训练格式:
现在labelme2yolov7seg.py文件中修改自己的类别名,然后
运行命令:
python labelme2yolov7seg.py --labelme_dataset_dir dataset --train_val_ratio 0.2 --ouput_dataset_dir yolov7_seg_dataset --image_name frame
其中,dataset文件夹中存放的是img和json混合数据,yolov7_seg_dataset文件夹是保存输出数据,
5、修改配置文件
5.1:在data文件夹下的custom.yaml文件中,修改自己数据路径和类别