【无标题】

最新推荐文章于 2024-10-04 17:46:34 发布

xiaobairen88

最新推荐文章于 2024-10-04 17:46:34 发布

阅读量718

点赞数

文章标签： xml 分类 python

本文链接：https://blog.csdn.net/xiaobairen88/article/details/121711628

版权


```python
import os
from shutil import copyfile


train_txt_path=""

val_txt_path=""

test_txt_path=""

all_img_path=""

train_img_path=""//训练图片存储位置

val_img_path=""

test_img_path=""

all_xml_path=""

train_xml_path=""

val_xml_path=""

test_xml_path=""


with open(train_txt_path,"r") as f:
    for line in f.readlines():
        line=line.strip('\n')
        img_path=all_img_path+"/"+line+".jpg"
        copyfile(img_path,train_img_path+"/"+line+".jpg")
        xml_path=all_xml_path+"/"+line+".xml"
        copyfile(xml_path,train_xml_path+"/"+line+".xml")
    

with open(val_txt_path,"r") as f:
    for line in f.readlines():
        line=line.strip('\n')
        img_path=all_img_path+"/"+line+".jpg"
        copyfile(img_path,val_img_path+"/"+line+".jpg")
        xml_path=all_xml_path+"/"+line+".xml"
        copyfile(xml_path,val_xml_path+"/"+line+".xml")
    
    
with open(text_txt_path,"r") as f:
    for line in f.readlines():
        line=line.strip('\n')
        img_path=all_img_path+"/"+line+".jpg"
        copyfile(img_path,test_img_path+"/"+line+".jpg")
        xml_path=all_xml_path+"/"+line+".xml"
        copyfile(xml_path,test_xml_path+"/"+line+".xml"))