赞
踩
源码: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文件对应
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。