网上有几个帖子都写的很糊,而且目录结构不清晰,也只支持单张截取,因为本人要制作数据集,所以不可能单个剪。所以编写此类,循环剪裁,切割大尺寸的tiff或者其他格式图片。
我是写java,感觉python的语法一言难用尽,所以在生成唯一影像上纠结半天,因为Python的int和str无法直接拼接成str,imwrite(写入函数)又要求地址必须str .java是可以的直接拼接的,想不清设计者的脑回路,难道我int拼str,还有可能想要int类型?
目录结构:
pudowntif用来放大tiff文件,outputtif,用来存放生成文件,就这两个文件夹,结构简单点,有一个帖子嵌套好几层不知道那个博主怎么写的。
实现部分:
获取自己想要切割的tiff的size,打开tif后在右上角我是1838*2258
import os
import cv2
'''
裁剪tiff图为(224,224)
'''
#source_dir = r'C:\Users\admin\PycharmProjects\untitled\putdowntif' 很多tiff需要分割的情况用
target_dir = r'C:\Users\admin\PycharmProjects\untitled\outputtif'
# 如果目标目录不存在的话,进行目录的新建
if not os.path.exists(target_dir):
os.makedirs(target_dir)
T = os.listdir(source_dir)
#我这里用单个tiff举例子
images = os.listdir(r'C:\Users\admin\PycharmProjects\untitled\putdowntif')
for image in images:
i=0 #标记图片的纵轴位置
j=0 #标记图片的横轴位置
c=0 #用来做唯一的输出名拼接,如果没有你每次生成的tif就会覆盖上一次的tif
while((i+128)<2258):# 2285横轴size
image_name = os.path.join(source_dir, image)
img = cv2.imread(image_name)
while((i+128)<2258 and (j+128)<1858):#1858纵轴size
c = c + 1
res = img[i:i+128, j:j+128] #你想要切割的每个模块大小,我选的是128*128 可以任意设置
j = j+128
print(res.shape)
image_src = os.path.join(target_dir, str(c)+image)#str(c)+image拼接的顺序不能调换
print(image_src+str(c)
cv2.imwrite(image_src,res) # 图片存储
print("finish" + " " + image_src)
int(c)
i = i+128
j = 0
生成结果:
实际用的就只是用的矩阵遍历的方法来切割