Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-25 12:39:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

add compatibility for torchvision>0.10.0

Browse Source 
add compatibility for torchvision>0.10.0
        Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/12292460
This commit is contained in:
xianzhe.xxz
2023-04-11 19:59:59 +08:00
committed by yuze.zyz
parent 2c15888084
commit 2fe3812f98
1 changed files with 19 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
modelscope/models/cv/tinynas_detection/damo/augmentations/box_level_augs/geometric_augs.py
View File

@@ -73,14 +73,25 @@ def _geometric_aug_func(x,
labels_new.append(labels[i])
else:
x_crop = transforms.functional.to_pil_image(x_crop.cpu())
x_crop = transforms.functional.affine(
x_crop,
angle,
translate,
scale,
shear,
resample=2,
fillcolor=tuple([int(i) for i in pixel_mean]))
try:
x_crop = transforms.functional.affine(
x_crop,
angle,
translate,
scale,
shear,
resample=2,
fillcolor=tuple([int(i) for i in pixel_mean]))
except Exception:
x_crop = transforms.functional.affine(
x_crop,
angle,
translate,
scale,
shear,
interpolation=2,
fill=tuple([int(i) for i in pixel_mean]))
x_crop = transforms.functional.to_tensor(x_crop).to(x.device)
x_crops.append(x_crop)
y = _transform(x, x_crops, boxes_crops, translate)