mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix format

pull/1956/head
felixfu 2024-11-01 23:37:26 +08:00
parent c0c65f2f4e
commit 2702f87d4e
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mmpretrain/datasets/transforms/processing.py
View File

@ -83,7 +83,8 @@ def register_vision_transforms() -> List[str]:
if inspect.isclass(_transform) and callable(
_transform) and not isinstance(_transform, (EnumMeta)):
TRANSFORMS.register_module(
module=lambda name=module_name: TorchVisonTransformWrapper(transform=torchvision.transforms[name], name=name),
module=lambda name=module_name: TorchVisonTransformWrapper(
transform=torchvision.transforms[name], name=name),
name=f'torchvision/{module_name}')
vision_transforms.append(f'torchvision/{module_name}')
return vision_transforms