We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7e6bfad commit 0379038Copy full SHA for 0379038
torchvision/transforms/v2/_color.py
@@ -8,6 +8,7 @@
8
9
from ._transform import _RandomApplyTransform
10
from ._utils import query_chw
11
+from .functional._utils import is_cvcuda_tensor
12
13
14
class Grayscale(Transform):
@@ -265,6 +266,8 @@ class RandomEqualize(_RandomApplyTransform):
265
266
267
_v1_transform_cls = _transforms.RandomEqualize
268
269
+ _transformed_types = _RandomApplyTransform._transformed_types + (is_cvcuda_tensor,)
270
+
271
def transform(self, inpt: Any, params: dict[str, Any]) -> Any:
272
return self._call_kernel(F.equalize, inpt)
273
0 commit comments