深度学习人工智能Paper年度会员(NLP方向)前沿版
# 使用图片增广,裁剪,翻转,变换颜色等# 这里需要注意的是,这里是语义分割,如果我们对图片进行了裁剪,那么我们也需要对他的label和边缘框进行裁剪# 参数列表(需要裁剪的img,需要裁剪的label,目标高,目标宽)def voc_rand_crop(feature, label,height,width):
"""随机裁剪特征和标签图像"""
# 获取随即裁剪的参数
rect = torchvision.transforms.RandomCrop.get_params(feature,(height,width))
# 按照刚设定的值,裁剪特征
feature = torchvision.transforms.functional.crop(feature, *rect)
# 按照刚设定的值,裁剪标签
label = torchvision.transforms.functional.crop(label, *rect)
return feature, label
imgs = []