车牌分类任务(三、批量修改车牌名称)

本文代码用于批量修改车牌名称!!!

# -*- coding=utf-8 -*-
import os

def img_rename(imgs_path):
    imgs_name = os.listdir(imgs_path)
    i = 0
    for img_name in imgs_name:
        if img_name.endswith('.jpg'):
            old_name = os.path.join(os.path.abspath(imgs_path), img_name)
            # 类别+图片编号    format(str(i),'0>3s') 填充对齐
            new_name = os.path.join(os.path.abspath(imgs_path), 'class_name' + format(str(i),'0>3s') + '.jpg')
            os.rename(old_name, new_name)
            i = i + 1
if __name__ == '__main__':
    imgs_path = 'read_img_file_path/'
    img_rename(imgs_path)

万一爬取得图片或搜集的图片中混入了灰度图,之后的神经网络在训练过程中将会遇到问题,所以可将此类型的图先找出来,再做下一步调整!!!

from PIL import Image
import  numpy as np
import os

def read_img_label(imgs_path):
    # 获得图片路径和类别
    for label_name in os.listdir(imgs_path):
        for img_name in os.listdir(imgs_path+label_name):
            img_path = imgs_path+label_name+'/'+img_name
            img = Image.open(img_path)
            if len(np.array(img).shape) == 2:
                print("灰度图路径:",img_path)

if __name__=='__main__':
    imgs_path = 'read_img_path/'
    read_img_label(imgs_path)

猜你喜欢

转载自blog.csdn.net/attitude_yu/article/details/81097133
今日推荐