近期在整OpenCV的分类器训练,弄了个批量修改图片格式的脚本分享下,转载请注明出处!!
# -*- coding: utf-8 -*-
# !/usr/bin/env python
# author :dean su
# date: 2018-10-16
# version: 0.1
import cv2,os,shutil
path="D:\\AI\\pic\\" #读取图片文件路径
path_="D:\\AI\\pictrain\\" #中转文件路径
_path="D:\\AI\\SN\\posdata\\" #存储修改后图片路径
if os.path.exists(path_) == False:
os.makedirs(path_)
o=os.listdir(path)
for n,a in enumerate(o):
n=n+1
n=str(n)
new_path=os.path.join(path+a)
new_path_=path_+n+".jpg"
print(new_path_)
os.rename(new_path,new_path_)
oo=os.listdir(path_)
for n,a in enumerate(oo):
new_path=os.path.join(path_+a)#获取文件
print("读取文件路径:",new_path)
image1=cv2.imread(new_path)
n=n+1
n=str(n)
image2=cv2.resize(image1,(40,40),interpolation=cv2.INTER_CUBIC)#修改尺寸40*40
gray=cv2.cvtColor(image2,cv2.COLOR_BGR2GRAY)#转为灰阶图像
path=os.path.join(_path+n+".jpg")
print("生成文件:",path)
cv2.imwrite(path,gray)