将文件夹内含有对应编号的数据输出

最新推荐文章于 2025-01-08 21:42:16 发布

Fly*Boy

最新推荐文章于 2025-01-08 21:42:16 发布

阅读量233

点赞数

CC 4.0 BY-SA版权

分类专栏：学习

本文链接：https://blog.youkuaiyun.com/mvpboy123/article/details/107301798

学习专栏收录该内容

32 篇文章

订阅专栏

本教程介绍了一种使用Python脚本从指定目录下筛选并复制与训练标签对应的图像数据的方法。通过遍历文件夹，该脚本能够根据训练标签将特定的.png文件从源目录复制到目标目录，适用于图像数据集的预处理阶段。

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

根据train_label，将data中与label相对应的数据输出到train_data
39是path的字符串个数，6248是文件个数

import os
import os.path
import shutil
import glob

path = "C:/Users/User/Desktop/colon/train_label/"
img_path = glob.glob(path+"/*.png")
for i in range(6248):
    # print(img_path[i][39:])
    print(i)
    data_path = "C:/Users/User/Desktop/colon/data/"+img_path[i][39:].replace('mask','countour')
    new_path = "C:/Users/User/Desktop/colon/train_data/"+img_path[i][39:].replace('mask','countour')
    # print(new_path)
    shutil.copyfile(data_path,new_path)