python 批量修改图片的名字

最新推荐文章于 2024-07-26 19:00:00 发布

夏天的欢

最新推荐文章于 2024-07-26 19:00:00 发布

阅读量4.5k

点赞数 3

分类专栏： python tool 文章标签： python opencv 批量修改图片名称

本文链接：https://blog.csdn.net/qq_36414085/article/details/105369958

版权

python tool 专栏收录该内容

13 篇文章 1 订阅

订阅专栏

文章目录

情况1：数据都在一个文件夹内
情况二：不同类别的数据在不同的文件夹内

情况1：数据都在一个文件夹内

修改前：
在这里插入图片描述
修改后：

代码：

import os
import cv2 as cv

image_path = 'C:/Users/huan4/Desktop/Python/TensorFlow/train_ourself_data/01_classification_network/data/train/'

for file in os.listdir(image_path):
    name = file.split(sep='_')
    if name[0] == '0':
        new_image = image_path + 'airplane' + '_' + name[1]
        scr = cv.imread(image_path + file)
        cv.imwrite(new_image, scr)
    elif name[0] == '1':
        new_image = image_path + 'car' + '_' + name[1]
        scr = cv.imread(image_path + file)
        cv.imwrite(new_image, scr)
    elif name[0] == '2':
        new_image = image_path + 'bird' + '_' + name[1]
        scr = cv.imread(image_path + file)
        cv.imwrite(new_image, scr)

情况二：不同类别的数据在不同的文件夹内

在这里插入图片描述

修改后：

import os
import cv2 as cv

data_path = 'C:/Users/huan4/Desktop/Python/TensorFlow/train_ourself_data/cat_and_dog/data/'
save_path = 'C:/Users/huan4/Desktop/Python/TensorFlow/train_ourself_data/cat_and_dog/train/'

for file in os.listdir(data_path):
    image_file_path = data_path + file + '/'
    i = 0
    for image_name in os.listdir(image_file_path):
        scr = cv.imread(image_file_path + image_name)
        dst = cv.resize(scr, (600, 800), interpolation=cv.INTER_CUBIC)  #如果图片太大，可以使用cv.resize成自己想要的大小
        new_image_name = save_path + file + '_' + str(i) + '.jpg'
        print(new_image_name)
        cv.imwrite(new_image_name, dst)
        i += 1