Skip to content

Commit 0379038

Browse files
committed
move transformed tpyes to equalie transform class
1 parent 7e6bfad commit 0379038

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchvision/transforms/v2/_color.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88

99
from ._transform import _RandomApplyTransform
1010
from ._utils import query_chw
11+
from .functional._utils import is_cvcuda_tensor
1112

1213

1314
class Grayscale(Transform):
@@ -265,6 +266,8 @@ class RandomEqualize(_RandomApplyTransform):
265266

266267
_v1_transform_cls = _transforms.RandomEqualize
267268

269+
_transformed_types = _RandomApplyTransform._transformed_types + (is_cvcuda_tensor,)
270+
268271
def transform(self, inpt: Any, params: dict[str, Any]) -> Any:
269272
return self._call_kernel(F.equalize, inpt)
270273

0 commit comments

Comments
 (0)