mask rcnn bencmark pytorch自定义数据集的方法

最新推荐文章于 2024-09-01 15:36:46 发布

hiudawn

最新推荐文章于 2024-09-01 15:36:46 发布

阅读量1.5k

点赞数

分类专栏： deeplearning

本文链接：https://blog.csdn.net/hiudawn/article/details/88812979

版权

前言

参考代码： mask rcnn benchmark

数据集来源：津南数字制造算法挑战赛【赛场二】初赛

这个代码不能直接运行，仅仅提供参考，本人也仅仅是接触检测不到一个礼拜，如果有什么疑问欢迎在讨论区交流。

1、数据解读

数据集训练train_no_poly.json的格式，类coco风格

import json
with open('../train_no_poly.json', 'r') as f:
    data = json.load(f)

print(data.keys())
>>> dict_keys(['info', 'licenses', 'categories', 'images', 'annotations'])

print(data['info'])
>>> {
   'description': 'XRAY Instance Dataset ', 'url': '', 'version': '0.2.0', 'year': 2019, 'contributor': 'qianxiao', 'date_created': '2019-03-04 08:52:50.852455'}

print(data['licenses'])
>>> [{
   'id': 1, 'name': 'Attribution-NonCommercial-ShareAlike License', 'url': ''}]

print(data['categories'])
>>> [{
   'id': 1, 'name': '铁壳打火机', 'supercategory': 'restricted_obj'}, {
   'id': 2, 'name': '黑钉打火机', 'supercategory': 'restricted_obj'}, {
   'id': 3, 'name': '刀具', 'supercategory': 'restricted_obj'}, {
   'id': 4, 'name':