1.下载标注工具用于打标签
使用标注工具,后面会用到智能标注
点击 创建AI多边形后命令行就自动下载对应的模型
单机要选中的图像就行,就可以智能选中,双击设置标签
依次标注所有图片 ,最后保存成json格式的文件
2.使用labelme2yolo将json转为yolo格式的数据集
命令:
pip install labelme2yolo
labelme2yolo --json_dir="E:/python_code/ultralytics-8.2.74/ultralytics/segemntTest/tupian" --val_size=0.1 --test_size=0.2 --output_format="polygon"
tupian里面是左图的文件
3.修改data与net文件
主要是修改数据集
3.执行训练代码
yolo segment train data=data-seg.yaml model=net-seg.yaml epochs=100 imgsz=640
yaml文件的路径修改成自己的,最好是用左斜杠