python工具方法 39 大图裁剪为小图|小图还原成大图(含生成大图伪标签)
在进行遥感影像的处理中,通常都是几万x几万的大型影像,这给数据标注(图像尺寸过大使标注软件不能正常打开或过大给标注带来困难)、训练模型(只能使用小图)、测试图片(在小图上预测需要还原成大图)、生成伪标签(需要大图的伪标签,而不是小图的)都带来了困难。针对此进行,实现函数一:大图裁剪为小图文件,实现函数二:小图还原成大图文件。
这里的核心函数依赖https://hpg123.blog.csdn.net/article/details/124824892所实现,需要把以上链接中章节一中的全部代码保存为imgutils.py
1、大图裁剪为小图文件
完整代码如下所示,需要根据自身需求修改path(图片的父路径)、save_dir(小图文件的保存路径)、crop_size(小图的裁剪尺寸)、pad(裁剪小图时两个图之间的重叠区域)
import math,os
from imgutils import *
import numpy as np
from PIL