json 轉 mot17數(shù)據(jù)格式的實現(xiàn)代碼 (親測有效)

from PIL import Image
import os
import glob
import numpy as np
import 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#計算圖片總數(shù)
  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 #獲取路徑下文件個數(shù) 
  print('%s路徑下有%d個文件'%(L[0],num))
  return L ,num ,img_num


def 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()