当前位置:   article > 正文

yolov7_实例分割(有待完善)_yolov7数据集格式用于实例分割

yolov7数据集格式用于实例分割

源码:https://github.com/jinfagang/yolov7
需要的安装包:
–torch 1.11 (stable version)
–onnx
–alfred-py
–nbnb
–shapely
detectron2(注意与torch、cuda、python版本匹配)https://github.com/facebookresearch/detectron2

一、制作数据集(参考coco数据集制作)

这是自己的数据集生成的instances_train2017.json文件
注意"categories",不同种类的"id"与"name"
在这里插入图片描述数据集格式:
train2017里面是训练图片
val2017里面是验证图片
在这里插入图片描述

二、修改部分代码

detectron2/data/builtin_meta.py

1、COCO_CATEGORIES = []中的"id"与"name"与自己的json文件对应
在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/凡人多烦事01/article/detail/320065
推荐阅读
相关标签
  

闽ICP备14008679号