指定图片名称分割数据集

最新推荐文章于 2025-12-16 10:44:31 发布

原创最新推荐文章于 2025-12-16 10:44:31 发布 · 155 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

图像分类专栏收录该内容

3 篇文章

订阅专栏

该代码段展示了如何使用Python的os和shutil模块来遍历指定目录下的图片文件，并根据文件名中的特定规律（数字尾数为1）将图片移动到新的目标路径下，用于数据集的整理和划分。

import os
import shutil
from shutil import copy, rmtree
import random

cwd = os.getcwd()
data_root = cwd
origin_path = os.path.join(data_root, "Anomaly")
new_picture_path = r"C:\Users\sun\Desktop\data1\val\Anomaly"
image = os.listdir(origin_path)
for images in image:
    origin_picture_path = os.path.join(origin_path, images)
    image_name = os.path.splitext(images)[0]
    image_number = int(image_name[8:])
    if image_number % 10 == 1:
        shutil.move(origin_picture_path, new_picture_path)