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

最新推荐文章于 2025-05-21 12:53:23 发布

cv.exp

最新推荐文章于 2025-05-21 12:53:23 发布

阅读量713

点赞数

CC 4.0 BY-SA版权

分类专栏： Python Deep Learning

本文链接：https://blog.youkuaiyun.com/forest_world/article/details/78254010

Deep Learning 同时被 2 个专栏收录

172 篇文章

订阅专栏

Python

13 篇文章

订阅专栏

本文介绍了一个Python脚本，用于自动创建VOC2007数据集所需的目录结构，并将图像文件和XML标注文件分别复制到JPEGImages和Annotations目录下。此外，还生成了trainval.txt和test.txt文件，列出了所有图片的基本名称。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os
import shutil
import xml.etree.ElementTree as ET

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