#-* coding -utf-8 -*
#这个脚本用来切割和拼接图片
#运行程序输入1则为切割图片模式
#输入其他数字则为拼接图片模式
import numpy as np
import os
import cv2
import PIL.Image as Image
#切图
def cut():
img = cv2.imread('./img/1792/org.tif', cv2.IMREAD_COLOR)
h = img.shape[0]
w = img.shape[1]
print(h)
print(w)
cv2.imshow('test',img)
cv2.waitKey()
cv2.destroyAllWindows()
size = 1792
# 开始切图 cut
h_step = img.shape[0] // size
w_step = img.shape[1] // size
h_rest = -(img.shape[0] - size * h_step)
w_rest = -(img.shape[1] - size * w_step)################256变64,size
image_list = []
predict_list = []
count = 0
# 循环切图
for h in range(h_step):
for w in range(w_step):
# 划窗采样
img_cut = img[h*size:(h+1)*size, w*size:(w+1)*size, :]
image_list.append(img_cut[:,:,1])
cv2.imwrite(('./img/cut/' + '%3d_sat.jpg' % count), img_cut)
count += 1
#拼接
def splice():
IMAGES_PATH = './img/cut/'
IMAGES_FORMAT = '.png'
IMAGE_SIZE = 256
IMAGE_ROW = 8
IMAGE_COLUMN = 8
IMAGE_SAVE_PATH = './img/cut/final.png'
# 获取图片集地址下的所有图片名称
filelist = os.listdir(IMAGES_PATH)
image_names = np.array([file for file in filelist if file.endswith('.png')], dtype=object)
# 简单的对于参数的设定和实际图片集的大小进行数量判断
if len(image_names) != IMAGE_ROW * IMAGE_COLUMN:
raise ValueError("合成图片的参数和要求的数量不能匹配!")
# 定义图像拼接函数
def image_compose():
to_image = Image.new('RGB', (IMAGE_COLUMN * IMAGE_SIZE, IMAGE_ROW * IMAGE_SIZE)) #创建一个新图
# 循环遍历,把每张图片按顺序粘贴到对应位置上
for y in range(1, IMAGE_ROW + 1):
for x in range(1, IMAGE_COLUMN + 1):
from_image = Image.open(IMAGES_PATH + image_names[IMAGE_COLUMN * (y - 1) + x - 1]).resize(
(IMAGE_SIZE, IMAGE_SIZE),Image.ANTIALIAS)
to_image.paste(from_image, ((x - 1) * IMAGE_SIZE, (y - 1) * IMAGE_SIZE))
return to_image.save(IMAGE_SAVE_PATH)
image_compose() #调用函数
model = int(input())
if model == 0:
cut()
else:
splice()
python 切割和拼接图片
最新推荐文章于 2024-04-19 16:44:29 发布