将文件夹中的文件名输出到txt文件

winnerziqi

于 2021-03-31 10:34:59 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： data augument 文章标签： xml zip linux powershell logback

本文链接：https://blog.csdn.net/wty98wzq/article/details/115344631

data augument 专栏收录该内容

14 篇文章

订阅专栏

import os
import random
from utils import mkdir

xmlfilepath = r"D:\Catalogue\master\generate_data\paste_mixup_30\VOC2007_multmixup_3\Annotations"   

txtsavepath = r'D:\Catalogue\master\generate_data\paste_mixup_30\VOC2007_multmixup_3\ImageSets\Main'
mkdir(txtsavepath)
#mkdir(picsavepath)
# 得到所有xml的数目
total_xml = os.listdir(xmlfilepath) 
num = len(total_xml)

ftrainval = open(txtsavepath+'/trainval.txt', 'w')
for i in range(num):
    name = total_xml[i][:-4]+'\n'
    ftrainval.write(name)
ftrainval.close()