python实现文件后缀名批量转换

dat_stick

于 2022-02-04 20:33:39 发布

阅读量800

点赞数

CC 4.0 BY-SA版权

文章标签： python 开发语言后端

本文链接：https://blog.youkuaiyun.com/dat_stick/article/details/122785598

该脚本用于批量处理'dataset/train/bees_image'目录下的图像，根据目标目录名创建标签文件。它首先获取图像路径，然后从目标目录名中提取标签，为每个图像创建一个.txt文件，写入相应的标签。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

root_dir = 'dataset/train'
target_dir = 'bees_image'
img_path = os.listdir(os.path.join(root_dir, target_dir))
label = target_dir.split('_')[0]
out_dir = 'bees_label'
for i in img_path:
    file_name = i.split('.jpg')[0]
    with open(os.path.join(root_dir, out_dir, "{}.txt".format(file_name)), 'w') as f:
        f.write(label)