diff --git a/rtdetr_pytorch/src/data/transforms.py b/rtdetr_pytorch/src/data/transforms.py
index 3fd3945cb..aab827541 100644
--- a/rtdetr_pytorch/src/data/transforms.py
+++ b/rtdetr_pytorch/src/data/transforms.py
@@ -140,3 +140,5 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
 
         return inpt
 
+    def transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
+        return self._transform(inpt, params)
diff --git a/rtdetrv2_pytorch/src/data/transforms/_transforms.py b/rtdetrv2_pytorch/src/data/transforms/_transforms.py
index 53840c30d..143d3f3b7 100644
--- a/rtdetrv2_pytorch/src/data/transforms/_transforms.py
+++ b/rtdetrv2_pytorch/src/data/transforms/_transforms.py
@@ -112,6 +112,9 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
 
         return inpt
 
+    def transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
+        return self._transform(inpt, params)
+
 
 @register()
 class ConvertPILImage(T.Transform):
@@ -133,4 +136,7 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
 
         inpt = Image(inpt)
 
-        return inpt
\ No newline at end of file
+        return inpt
+
+    def transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
+        return self._transform(inpt, params)
\ No newline at end of file