最近要做图片的预处理工作,目的是把原图像的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.split(".")[0] for s in mg if not s.split(".")[0] == "Thumbs"]
#print("imgs:", imgs)
per_img_path = [i.replace('db-before', 'db-after') for i in imgs]
per_img_path=[i.replace('\\', '/') for i in per_img_path]