深度学习—图片尺寸及标签的设置

最新推荐文章于 2023-05-26 16:49:35 发布

qq_29934825

最新推荐文章于 2023-05-26 16:49:35 发布

阅读量3.1k

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_29934825/article/details/85157811

本文介绍了一种使用Python批量处理图片的方法，包括调整图片尺寸和进行标签化重命名，适用于图像数据预处理阶段，为后续的图像识别或分类任务做准备。

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

import numpy as np
import imageio
from PIL import Image
import os.path
import PIL
import glob
'''
文件夹需要提前创建好
'''
def rename(path):
    map_path = [str(path) + str(x) for x in os.listdir(path) if os.path.isfile(str(path) + str(x))]
    for idx, folder in enumerate(map_path):
        image = np.array(PIL.Image.open(folder))
        imageio.imwrite('E:/三类图片/狗_标签化/' + '2' + '_' + str(idx) + '.jpg', image)  # 要保存的标签化地址

def convertjpg(jpgfile, outdir, width=64, height=64):
    img = Image.open(jpgfile)
    try:
        new_image = img.resize((width, height), Image.BILINEAR)
        new_image.save(os.path.join(outdir, os.path.basename(jpgfile)))
    except Exception as e:
        print(e)

def resize(path):
    for jpgfile in glob.glob("E:/三类图片/狗/*.jpg"):
        convertjpg(jpgfile, path)

if __name__ == '__main__':
    path = r"E:/三类图片/狗_尺寸化/"
    resize(path)  # 先进行尺寸的调整
    rename(path)  # 再进行有标签重命名