Python脚本实现批量生成带标签图片列表txt文件
发布时间：2026-10-13 |
5人浏览过
本文介绍如何使用Python脚本批量处理图片文件夹，自动生成包含图片名称及对应标签的txt文件。该方法适用于深度学习数据预处理场景，通过遍历指定目录下的图片文件，按顺序写入文件名与标签信息，简化标注流程，提高数据处理效率。
在深度学习任务中，通常需要将图片与其对应的标签整理为特定的文本列表格式（如 xxxlist.txt）。为此，这里提供一个基础的Python脚本，用于自动化生成此类包含标签信息的图片列表文件。
import os
def generate(dir,label):
files = os.listdir(dir)
files.sort()
print '****************'
print 'input :',dir
print 'start...'
listText = open(dir+'\\'+'list.txt','w')
for file in files:
fileType = os.path.split(file)
if fileType[1] == '.txt':
continue
name = file + ' ' + str(int(label)) +'\n'
listText.write(name)
listText.close()
print 'down!'
print '****************'
if __name__ == '__main__':
generate('C:\\Users\\Desktop\\image',1)
执行上述脚本后，系统会在指定的 image 文件夹目录下生成一个 list.txt 文件。该文件中的每一行都会记录一张图片的名称，并附带标签值 1，从而完成图片与标签的绑定。
以上即为利用Python实现文件中图片生成带标签txt文件的具体方法。希望这段内容能为您提供参考，如果您觉得有用，欢迎继续支持本站获取更多技术分享。
本站声明
本文内容仅供参考，不代表本站观点。转载或引用请注明出处。如发现内容问题，请通过联系我们页面反馈。
