获取文件夹下图片的名字保存为txt文件

最新推荐文章于 2021-03-16 11:10:14 发布

GISer_精灵的光轨

最新推荐文章于 2021-03-16 11:10:14 发布

阅读量765

点赞数 1

分类专栏：机器学习

本文链接：https://blog.csdn.net/qq_34585338/article/details/90299401

版权

机器学习专栏收录该内容

19 篇文章 1 订阅

订阅专栏

import os
import random

xmlfilepath = 'E:\\LT0517\\label2'   #注意这里是image而非mask
txtsavepath = 'E:\\LT0517'
total_xml = os.listdir(xmlfilepath)

num=len(total_xml)
list = range(num)


trainval = random.sample(list, num)  

os.chdir('E:\\LT0517')   

ftrainval = open('name.txt', 'w')  # chuang jian yi ge txt wen jian

for i in list :
  name =total_xml[i][:-4] + '\n'
  ftrainval.write(name)
ftrainval.close()