本文代码用于批量修改车牌名称!!!
# -*- 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)