CUB 鸟类图像数据集,含有11788张图片,200个鸟类别,其中80%的图片的目标占比小于0.5。为方便合成任务,需要提高目标在全图的占比,为此进行如下预处理。
预处理:图像裁剪,裁剪多余背景区域,保证bbox划定的鸟的区域占图像比例不低于75% (bbox最大边至少为图像最大边的0.75倍)
import numpy as np
def custom_crop(img, bbox):
# bbox = [x-left, y-top, width, height]
imsiz = img.shape # [height, width, channel]