深度學習人工智能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 = []
標簽: