mirrors
/
fast-reid
mirror of https://github.com/JDAI-CV/fast-reid.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

change interpolation from BICUBIC to BILINEAR

pull/608/head
zuchen.wang 2021-11-12 15:22:45 +08:00
parent d543b80dcf
commit 88cc245847
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fastreid/data/transforms/build.py
View File

@ -65,11 +65,11 @@ def build_transforms(cfg, is_train=True):
if size_train[0] > 0: if size_train[0] > 0:
res.append(T.Resize(size_train[0] if len(size_train) == 1 else size_train, res.append(T.Resize(size_train[0] if len(size_train) == 1 else size_train,
interpolation=InterpolationMode.BICUBIC)) interpolation=InterpolationMode.BILINEAR))
if do_crop: if do_crop:
res.append(T.RandomResizedCrop(size=crop_size[0] if len(crop_size) == 1 else crop_size, res.append(T.RandomResizedCrop(size=crop_size[0] if len(crop_size) == 1 else crop_size,
interpolation=InterpolationMode.BICUBIC, interpolation=InterpolationMode.BILINEAR,
scale=crop_scale, ratio=crop_ratio)) scale=crop_scale, ratio=crop_ratio))
if do_pad: if do_pad:
res.extend([T.Pad(padding_size, padding_mode=padding_mode), res.extend([T.Pad(padding_size, padding_mode=padding_mode),
@ -96,8 +96,8 @@ def build_transforms(cfg, is_train=True):
if size_test[0] > 0: if size_test[0] > 0:
res.append(T.Resize(size_test[0] if len(size_test) == 1 else size_test, res.append(T.Resize(size_test[0] if len(size_test) == 1 else size_test,
interpolation=InterpolationMode.BICUBIC)) interpolation=InterpolationMode.BILINEAR))
if do_crop: if do_crop and size_test == crop_size:
res.append(T.CenterCrop(size=crop_size[0] if len(crop_size) == 1 else crop_size)) res.append(T.CenterCrop(size=crop_size[0] if len(crop_size) == 1 else crop_size))
res.append(ToTensor()) res.append(ToTensor())
return T.Compose(res) return T.Compose(res)