数据集处理——生成图片的txt路径文件

最新推荐文章于 2025-06-24 22:04:10 发布

原创最新推荐文章于 2025-06-24 22:04:10 发布 · 2.2k 阅读

19 ·

CC 4.0 BY-SA版权

文章标签：

#数据集处理

数据集解析专栏收录该内容

9 篇文章

订阅专栏

本文介绍了一种从COCO数据集中特定文件夹读取所有图像名称，并将这些图像的完整路径写入txt文件的方法。此过程通过Python脚本实现，利用os模块的listdir函数获取文件夹内所有图像文件名，然后逐条写入指定的txt文件中。

部署运行你感兴趣的模型镜像

1.根据图像所在的文件夹data_path

2.生成图像路径的txt文件

import os

data_path = 'COCO/coco_bicycle_new/2014_train/JPEGImages'
img_names = os.listdir(data_path)

list_file = open('del_coco_2014_train.txt', 'w')
for img_name in img_names:

    list_file.write('/media/voc_coco_bicycle/coco_bicycle_new/2014_train/JPEGImages/%s\n'%img_name)    

list_file.close()