[Fix] Fix PadToWidth register

pull/1178/head
wangxinyu 2022-06-07 03:47:54 +00:00 committed by gaotongxiao
parent cb8f980bae
commit 34a96a8b87
1 changed files with 1 additions and 0 deletions

View File

@ -1245,6 +1245,7 @@ class RescaleToHeight(BaseTransform):
return repr_str
@TRANSFORMS.register_module()
class PadToWidth(BaseTransform):
"""Only pad the image's width.