在mmdetection中推理无标注图像文件并输出检测结果(json文件)

使用不带标签的数据集对模型进行检测

不带标签图像数据集转化

MMDetection 支持在不使用 ground-truth 标注的情况下对模型进行测试,这需要用到 CocoDataset。如果你的数据集格式不是 COCO 格式的,请将其转化成 COCO 格式。如果你的数据集格式是 VOC 或者 Cityscapes,你可以使用 tools/dataset_converters 内的脚本直接将其转化成 COCO 格式。如果是其他格式,可以使用 images2coco 脚本进行转换。

python tools/dataset_converters/images2coco.py \
    ${
   IMG_PATH} \
    ${
   CLASSES} \
    ${
   OUT} \
    [--exclude-extensions]

参数

  • IMG_PATH是未标注的图像路径
  • CLASSES是一个文本文件,里面包含你的类别名,每行存储一个类别名
  • OUT:输出的json文件名,默认路径和IMG_PATH是同一路径
  • exclude-extensions:待排除文件后缀名
#例如:
python tools/dataset_converters/images2coco.py \
    image \
    label.txt \
    test \

生成图片的json后,json里的图片顺序可能和文件夹中的图片顺序,不一致,使用以下代码排序

import json

# 读取原始JSON文件
with open('data/annotations/sheep.json', 'r') as input_file:
    data = json.load(input_file)

# 修改id和file_name的对应关系
for item in data["images"]:
    id = item["id"]
    item["file_name"] =
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值