要求:对路径下所有图片遍历裁剪,保存其裁剪后的图片、npy和mat格式数据。
实现代码:
# -*- coding:utf-8 -*-
# @zhygou@163.com
import glob
import os
import cv2
import numpy as np
from multiprocessing import Pool
from PIL import Image
import scipy.io as io
def cut(img_dir, output_img, output_npy, output_mat, patch_size_w, patch_size_h, stride_w, stride_h):
"""
传入图片的路径,遍历切割;
img_dir 为图片待切割图片路径;
out_img:切割好的img的输出路径;
output_npy:输出npy
output_mat:输出mat数据的路径
patch_size_w:切割的图像宽度
patch_size_h:切割的图像的高度
stride_w:横向移动大小
stride_h:高度移动大小
"""
img_dir = img_dir
file_list = glob.glob(img_dir + '*.jpg')
print len(file_list)
for i in range(len(file_list)):
img = Image.open(file_list[i])
#img = cv2.imread(),cv2生成的是numpy
#AttributeError: 'numpy.ndarray' objec