yolov7_实例分割(有待完善)

本文介绍了如何利用Yolov7进行实例分割,包括所需安装的依赖如torch和detectron2,并强调了数据集的制作过程,特别是参照COCO数据集格式创建自定义数据集的实例,以及修改detectron2代码以适配自定义类别。
摘要由CSDN通过智能技术生成

源码: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文件对应

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值