COCO格式转yolov5训练格式

桃扇骨 2022-10-08 15:19 187阅读 0赞

import json
    
    with open('/home/lixuan/fromLX/场景识别/新场景样本/休闲细支新主题陈列/场景识别/训练样本初始数据/1_json/annotations/instances.json') as f:
        Json = json.load(f)
        annotations = Json['annotations']
        images = Json['images']
        image_id_name_dict = {}
        image_id_width_dict = {}
        image_id_height_dict = {}
        for image in images:
            image_id_name_dict[image['id']] = image['file_name']
            image_id_height_dict[image['id']] = image['height']
            image_id_width_dict[image['id']] = image['width']
        for annotation in annotations:
            id = annotation['id']
            bbox = annotation['bbox']
            x,y,w,h = bbox
            # x1 = x - w / 2
            # y1 = y - h / 2
            # x2 = x1 + w
            # y2 = y1 + h
            width = image_id_width_dict[id]
            height = image_id_height_dict[id]
            x = str(x / width)
            y = str(y / height)
            w = str(w / width)
            h = str(h / height)
            with open('/home/lixuan/fromLX/场景识别/新场景样本/休闲细支新主题陈列/场景识别/训练样本初始数据/1_json/labels/{}.txt'.format(image_id_name_dict[id].split('.')[0]),'w') as f:
                f.write('0 ' + x + ' ' + y + ' ' + w + ' ' + h + '\n')

发表评论取消回复

表情：

评论列表（有 0 条评论，187人围观）

还没有评论，来说两句吧...

相关阅读

相关 COCO数据格式说明

COCO数据格式说明 > 标注工具推荐：https://github.com/wkentaro/labelme > > labelme自带了转换VOC和COCO

淩亂°似流年/ 2023年01月17日 09:28/ 0 赞/ 339 阅读

相关 0032-目标检测-coco格式转voc格式

import os import numpy as np import codecs import json from glob imp

旧城等待，/ 2023年01月11日 08:47/ 0 赞/ 144 阅读

相关 COCO转YoloV5数据集，适用于YoloV5、ScaledYoloV4

import json import glob import os import shutil from pathlib import

旧城等待，/ 2022年12月29日 09:07/ 0 赞/ 160 阅读

相关 YOLOV5训练

简介 `YOLOv4`还没有退热，`YOLOv5`就已经来了！ 6月9日，`Ultralytics`公司开源了`YOLOv5`，离上一次`YOLOv4`发布不到50天，不过

Dear 丶/ 2022年12月13日 04:08/ 0 赞/ 185 阅读

相关 COCO格式转yolov5训练格式

import json with open('/home/lixuan/fromLX/场景识别/新场景样本/休闲细支新主题陈列/场景识别/训练样本初始

桃扇骨/ 2022年10月08日 15:19/ 0 赞/ 188 阅读

相关 datatrucks转coco格式

import json import os import cv2 import numpy as np with

怼烎@/ 2022年09月07日 14:57/ 0 赞/ 170 阅读

相关使用Yolov5进行目标检测并训练自己的VOC格式数据集

使用Yolo v5进行目标检测并训练自己的VOC格式数据集一、前提准备二、下载代码及配置环境三、下载预训练模型四、预测五、训练

桃扇骨/ 2022年09月02日 08:40/ 0 赞/ 313 阅读

相关 rtf格式转word格式

有时候，需要在Mac下面完成工作，将需要的信息保存到文本。可是，就算是打开的txt文件，最后居然没有保存选项，只能另存为，而且，还不能存成txt格式。那，我们从mac那里拷下来

野性酷女/ 2022年02月19日 10:39/ 0 赞/ 404 阅读

相关 Pascal VOC数据格式转COCO数据格式脚本（Object Detection）

1. 前言 1.1 COCO数据集 COCO的全称是Common Objects in COntext，是微软团队提供的一个可以用来进行图像识别的数据集。MS CO

妖狐艹你老母/ 2022年01月30日 14:59/ 0 赞/ 479 阅读

相关将keras训练的h5模型转成pb格式

import os import tensorflow as tf from keras import backend as K import logging

骑猪看日落/ 2022年01月22日 01:19/ 0 赞/ 289 阅读