根据图片的命名来把图片移动到相应的label文件夹以cub数据集为例

最新推荐文章于 2025-01-02 07:30:00 发布

原创最新推荐文章于 2025-01-02 07:30:00 发布 · 561 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

那些年，我们写过的python脚本专栏收录该内容

11 篇文章

订阅专栏

本文介绍了一种使用Python脚本根据图片文件名将其自动归类到相应文件夹的方法，通过正则表达式匹配图片名中的标签，实现图片的智能分类。

需求：根据图片的命名来把图片移动到相应的Label文件夹

待移动的图片：(位于/Datasets/cub200/)
在这里插入图片描述
目标文件夹：(位于/Datasets/cub200/images/)

代码：

import os
import shutil
import re
source = '/Datasets/cub200/'
target = '/Datasets/cub200/images/'
imgfile = os.listdir(source)
targetfile = os.listdir(target)
for img in imgfile:
    if not os.path.isdir(img):
        img_path = os.path.join(source, img)
        (shotname, extension) = os.path.splitext(img)
        print(shotname)
        if extension == '.jpg':
            label = re.search(r'[a-zA-Z_]*[a-zA-Z]', shotname) 
            print(label)
            for label_dir in targetfile:
                label_target = re.search(r'[a-zA-Z_]+', label_dir)
                #print(label_target)
                if label_target.group() == label.group():
                print(label_dir)
                target_path = os.path.join(target, label_dir)
                print(target_path)
                shutil.copy(img_path, target_path)