最近要做图片的预处理工作,目的是把原图像的roi提取出来。这里使用python做图像的批量裁剪 ,代码如下:
# coding: utf-8
from PIL import Image
import os
import os.path
import numpy as np
import cv2
imgs = []
os_getcwd=os.getcwd().replace('\\','/')
dbDir = os_getcwd + "/db-before/"
print(os_getcwd)
people = os.listdir(dbDir)
people.sort()
print(people)
path_after = r'../db-after'
for person in people:
personDir = dbDir + person + "/"
hands = os.listdir(personDir)
#print('personDir:',personDir)
#print('hands:', hands)
for hand in hands:
os.makedirs('crop-image'+'./'+path_after+'./'+person+'./'+hand)
handDir = personDir + hand + "/"
mg = os.listdir(handDir)
mg.sort()
#print("mg:",mg)
imgs = imgs + [handDir + s.spl