import subprocess
import multiprocessing
import os
from os import path
import shutil
import argparse
import sys
from multiprocessing.dummy import Pool as ThreadPool
import sys
import time
f=open("LPP.txt","w+")#输出的文本,可改成json
sys.stdout=f
def getFile(Files_list):#获取文件夹
workpath=path.dirname(__file__) # path为文件夹路径
file_list = os.listdir(workpath) # file_list即为路径下文件夹名或文件名
for files in file_list:
fulldirct =os.path.join(workpath,files)
if os.path.isdir(fulldirct):
Files_list.append(files)
#print(Files_list)
def getPicture(path, list_name): #获取文件夹下的图片
for files in os.listdir(path):
file_path = os.path.join(path, files)
if os.path.splitext(file_path)[1]=='.jpg':
list_name.append(files)
#print(Picture_list)
def getTxt(path):#获取txt中的第一行数据
file_path = path+"\\"+"groundtruth_rect.txt"
with open(file_path, "r") as f:
line =f.readlines()
firstline=line[0].strip('\n')
sub_str = firstline.split('\t')
sub_str=[x for x in sub_str if x!='']
print("[",end="")
for s in range(0,len(sub_str)):
if s!=0:
print(", ",end="")
print(sub_str[s],end="")
print("]",end="")
#print(sub_str)
workpath=path.dirname(__file__)
#testpath=workpath+"\\"+"lp.txt"
#temp_file=open(testpath,'w')
Files_list=list()
getFile(Files_list)
#for File in Files_list:
print("{",end="")
for i in range(0,len(Files_list)):
Picture_list=list()
if i!=0:
print(", ",end="")
print("\"",end="")
print(Files_list[i],end="")
print("\": {\"video_dir\": \"",end="")
print(Files_list[i],end="")
print("\", \"img_names\": [",end="")
Filepath=os.path.join(workpath,Files_list[i])
getPicture(Filepath,Picture_list)
for j in range(0,len(Picture_list)):
if j!=0:
print(", ",end="")
print("\"",end="")
print(Files_list[i],end="")
print("/",end="")
print(Picture_list[j],end="")
print("\"",end="")
print("], \"init_rect\": ",end="")
#print("[",end="")
getTxt(Filepath)
print(", \"gt_rect\": [",end="")
for k in range(0,len(Picture_list)):
if k == 0:
getTxt(Filepath)
else:
print(", ",end="")
print("[0, 0, 0, 0]",end="")
print("]}",end="")
print("}",end="")
#for j in range(0,len(Picture_list)):