yolov7训练分割数据集_yolov7-segment 训练

基于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文件中，修改自己数据路径和类别