import cv2
import os
import matplotlib.pylab as plt
# 缩小图像
def shrink_DirImages(source_path,save_path):
if not os.path.exists(source_path):
return
if not os.path.exists(save_path):
os.makedirs(save_path)
sour_path = source_path
sav_path = save_path
for i in range(1,100):
source_path = os.path.join(source_path,str(i))
save_path = os.path.join(save_path,str(i))
for file in os.listdir(source_path):
img = plt.imread(os.path.join(source_path,file), 0)
print(file)
# if img == None:
# "Error: could not load image"
# os._exit(0)
# height, width = img.shape[:2]
size = (int(1024), int(1024))
shrink = cv2.resize(img, size, interpolation=cv2.INTER_AREA)
cv2.imwrite(os.path.join(save_path,file),shrink)
source_path = sour_path
save_path = sav_path
# def shrink_Images(source_path,save_path):
# if not os.path.exists(source_path):
# return
# if not os.path.exists(save_path):
# os.makedirs(save_path)
# for file in os.listdir(source_path):
# img = plt.imread(os.path.join(source_path, file), 0)
# print(file)
#
# size = (int(512), int(512))
# # size = (int(1024), int(1024))
# shrink = cv2.resize(img, size, interpolation=cv2.INTER_AREA)
# cv2.imwrite(os.path.join(save_path, file), shrink)
# 放大图像
def amplification_image():
fx = 1.6
fy = 1.2
# enlarge = cv2.resize(img, (0, 0), fx=fx, fy=fy, interpolation=cv2.INTER_CUBIC)
if __name__ == '__main__':
source_path = 'H:\\0\\test'
shrink_Images(source_path,source_path)