资源简介
这是一个代码,用于将精细标注的语义分割(多边形标注的json文件)标注转为目标检测框标注文件,并生成yolo需要的txt文件(需要的输出格式你们可以自己改)
代码片段和文件信息
import json
import os
from os import listdir getcwd
from os.path import join
import os.path
rootdir=‘/home/wang/下载/数据集/cityscapes/Images/train‘ #写自己存放图片的数据地址
def position(pos): #该函数用来找出xminyminxmaxymax即bbox包围框
x=[]
y=[]
nums=len(pos)
for i in range(nums):
x.append(pos[i][0])
y.append(pos[i][1])
x_max=max(x)
x_min=min(x)
y_max=max(y)
y_min=min(y)
b=(float(x_min)float(x_max)float(y_min)float(y_max))
return b
def convert(size box): #该函数将xminyminxmaxymax转为xywh中心点坐标和宽高
dw = 1./(size[0])
dh = 1./(size[1])
x = (box[0] + box[1])/2.0 - 1
y = (box[2] + box[3])/2.0 - 1
w = box[1] - box[0]
h = box[3] - box[2]
x = x*dw
w = w*dw
y = y*dh
h = h*dh
return (xywh)
def convert_annotation(image_id):
load_f=open(“./train/%s.json“%(image_id)‘r‘)#导入json标签的地址
load_dict = json.load(load_f)
out_file = open(‘./voc_type/train/%s.txt‘%(image_id) ‘w‘) #输出标签的地址
#keys=tuple(load_dict.keys())
w=load_dict[‘imgWidth‘] #原图的宽,用于归一化
h=load_dict[‘imgHeight‘]
#print(h)
objects=load_dict[‘objects‘]
相关资源
- keras上LSTM长短期记忆网络金融时序预
- 基于PCA实现鸢尾花数据集降维
- 深度学习 莫烦 Keras源代码
- BP神经网络马疝病数据集
- 用自己的数据制作python版本cifar10数据
- 合并BN层的python脚本
- 机器学习深度学习篇系列分享_超值
- 随机森林的代码实现和相应的数据集
- 使用trainId图片评估cityscapes数据集Io
- 《TensorFlow2深度学习》
- 深度学习视频教程,包括python入门,
- python三阶深度学习框架-Real-Time-Voice
-
xm
l_parse.py - CIFAR10数据集python版本
- Python-人工智能和基于机器学习的机器
- Apriori关联性分析python实现(含数据集
- 可直接运行版本python实现yolov3调用摄
- Deep Learning for Computer Vision with Python链
- KITTI数据集三维可视化代码
- 莫烦全部代码Reinforcement-learning-with-
- cifar-10-python.tar.gz的资源
- python全栈视频某智
- pima_data.csv印第安人糖尿病数据集
- YOLO_train.py
- 强化深度学习迷宫问题
- 基于PyTorch的深度学习技术进步
-
深度学习目标检测提取xm
l文件中的 - 从图像数据集读取图片并拼接成大图
- 深度学习入门:基于Python的理论与实
- 基于MTCNN实现制作脸部VOC格式数据集
评论
共有 条评论