Python Faster R-CNN深度学习训练数据处理

来源：互联网发布：伦敦金今晚数据编辑：程序博客网时间：2024/06/06 22:13

import osimport shutilimport xml.etree.ElementTree as ETimport pdb#shutil.rmtree(os.getcwd()+"/VOC2007"); os.mkdir (os.getcwd()+"/VOC2007");os.mkdir (os.getcwd()+"/VOC2007/Annotations");os.mkdir (os.getcwd()+"/VOC2007/ImageSets"); os.mkdir (os.getcwd()+"/VOC2007/ImageSets/Main");os.mkdir (os.getcwd()+"/VOC2007/JPEGImages"); my_files = os.listdir(os.getcwd()+"/");for files in my_files:    if "jpg" in files:        print(os.getcwd() + "/" + files)        print('\n')          print(os.getcwd()+"/VOC2007/JPEGImages/" + files)        #pdb.set_trace()         shutil.copyfile(os.getcwd() + "/" + files, os.getcwd()+"/VOC2007/JPEGImages/" + files)for files in my_files:    if "xml" in files:        shutil.copyfile(os.getcwd() + "/" + files, os.getcwd()+"/VOC2007/Annotations/" + files)  #my_files = os.listdir("jpg");#print(my_files)listFiles = []for files in my_files:    if "jpg" in files:        #print(files)        listFiles.append(files[:-4])     listFiles.sort()fo = open(os.getcwd()+"/VOC2007/ImageSets/Main/" + "trainval.txt", "w")for fileName in listFiles:    fo.write(fileName + "\n")    print(fileName)fo.close()  fo = open(os.getcwd()+"/VOC2007/ImageSets/Main/" + "test.txt", "w")for fileName in listFiles:    fo.write(fileName + "\n")    print(fileName)fo.close()

阅读全文

0 0