您当前的位置:首页 > IT编程 > python
| C语言 | Java | VB | VC | python | Android | TensorFlow | C++ | oracle | 学术与代码 | cnn卷积神经网络 | gnn | 图像修复 | Keras | 数据集 | Neo4j | 自然语言处理 | 深度学习 | 医学CAD | 医学影像 | 超参数 | pointnet | pytorch | 异常检测 | Transformers | 情感分类 | 知识图谱 |

自学教程:json 转 mot17数据格式的实现代码 (亲测有效)

51自学网 2021-10-30 22:49:05
  python
这篇教程json 转 mot17数据格式的实现代码 (亲测有效)写得很实用,希望能帮到您。

代码使用说明

1970-2270文件夹是保存图像和json文件(也就是需要进行转换的文件)
det文件夹是保存单个json对应的txt(因为np.savetxt函数只能进行单个数组的保存)
det.txt文件是整合det文件夹所有txt文件(mot17数据集格式)

在这里插入图片描述

完整代码

from PIL import Imageimport osimport globimport numpy as npimport json#读取图片,修改图片,另存图片def convertjpg(jpgfile,outdir,img_sum=0):  img=Image.open(jpgfile)#提取目录下所有图片  try:    img_sum=str('%08d'%img_sum)#图片保存成00000001格式    img.save(os.path.join(outdir)+img_sum+'.jpg')#保存到另一目录  except Exception as e:    print(e)#读取文件名def file_name(file_dir):    L=[]#保存文件名  img_num=0#计算图片总数  for root, dirs, files in os.walk(file_dir):    img_num=img_num+len(files)    one=os.path.basename(str(root))#获取路径最后的/或者/后的文件名    L.append(one)   num=len(L)-1 #获取路径下文件个数   print('%s路径下有%d个文件'%(L[0],num))  return L ,num ,img_numdef json2det(json_dir,out_dir,json_num):  with open(json_dir,'r') as f:    data_dict = json.load(f)  '''5行5个目标,6列(类别索引、框左上角x坐标、框左上角y坐标、框的宽、框的高、目标置信度100)'''  data=np.zeros([5,6])  for i in range(len(data_dict["shapes"])):#遍历一个json中的所有目标    points = np.array(data_dict["shapes"][i]["points"])    xmin=min(points[:,0])    ymin=min(points[:,1])    xmax=max(points[:,0])    ymax=max(points[:,1])        data[i,0]=json_num    data[i,1]=xmin    data[i,2]=ymin    data[i,3]=xmax-xmin    data[i,4]=ymax-ymin    data[i,5]=100  print(data)        a=np.linspace(-1,-1,len(data))  data=np.insert(data,1,a,axis=1)#行矩阵插入  a=a.reshape((len(a),1))  data=np.concatenate((data,a,a,a),axis = 1)#补充-1  print(data,'/n')      np.savetxt(out_dir,data,fmt="%d,%d,%.3f,%.3f,%.3f,%.3f,%.3f,%d,%d,%d",delimiter="/n")def main():    i=0  data_dir='E:/DL/CSDN-blog/json2mot17'  for jsonfile in glob.glob(data_dir+'/1970-2270/'+'*.json'):    i=i+1    json2det(jsonfile,data_dir+'/det/'+str(i)+'.txt',i)  print('det.txt生成完成')    det=open(data_dir+'/det.txt','w')  for txts in range(300):    print(txts)    one_det=open(data_dir+'/det/'+str(txts+1)+'.txt').read()    det.write(one_det)      det.close()  if __name__ == '__main__':  main()

代码执行结果

在这里插入图片描述

到此这篇关于json 转 mot17数据格式 (亲测有效)的文章就介绍到这了,更多相关json 转 mot17数据格式 内容请搜索51zixue.net以前的文章或继续浏览下面的相关文章希望大家以后多多支持51zixue.net!


python统计RGB图片某像素的个数案例
Python 实现图像逐像素点取邻域数据
万事OK自学网:51自学网_软件自学网_CAD自学网自学excel、自学PS、自学CAD、自学C语言、自学css3实例,是一个通过网络自主学习工作技能的自学平台,网友喜欢的软件自学网站。