您现在的位置是:首页 >技术教程 >opencv 裁剪超大 tif 图像制作深度学习数据集网站首页技术教程

opencv 裁剪超大 tif 图像制作深度学习数据集

Sunshine_xiaohao 2024-06-17 10:48:33
简介opencv 裁剪超大 tif 图像制作深度学习数据集


本文只是进行了简单的尝试,未单独处理最右侧、最下侧和右下角区域,且未设置重叠率
若使用 gdal 进行 tif 图像裁剪,可以参考 这里,其作者也是做遥感图像处理的,可关注他的其它文章

1. 尝试暴力裁剪(24 位)

def CropTif(src, dest, upperleft, lowerright, size):
    """
    本函数没有单独处理下方和右侧边缘区域,且由于用于水体提取,未涉及重叠裁剪
    :param src: 源图像地址
    :param dest: 目标文件夹地址
    :param upperleft: 要裁剪区域的左上角坐标
    :param lowerright: 要裁剪区域的右下角坐标
    :param size: 每次裁剪的大小
    """
    img = cv2.imread(src)
    hmin, wmin = upperleft
    hmax, wmax = lowerright
    filenumber = 1
    h = hmin
    while h <= hmax:
        w = wmin
        while w <= wmax:
            if h + size <= hmax and w + size <= wmax:
                cv2.imwrite(dest + f"/{filenumber}.tif", img[h:h+size, w:w+size])
                filenumber = filenumber + 1
            w = w + size
        h = h + size

我们的原图是 32 (RGBA) 位的图像,直接使用 cv2.imread(src) 方式进行图像读取会忽略 A (图像透明度),即出现下面的效果
在这里插入图片描述在这里插入图片描述

2. 带透明度读取 (32 位)

img = cv2.imread(src, -1)

在这里插入图片描述

3. 裁剪标签(8 位)

img = cv2.imread(src, 0)

在这里插入图片描述

风语者!平时喜欢研究各种技术,目前在从事后端开发工作,热爱生活、热爱工作。