Skip to content

Commit 8ec91b2

Browse files
authored
fix: #561 - NotImplementedError at transforms (#584)
1 parent 93d00c1 commit 8ec91b2

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

rtdetr_pytorch/src/data/transforms.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -140,3 +140,5 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
140140

141141
return inpt
142142

143+
def transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
144+
return self._transform(inpt, params)

rtdetrv2_pytorch/src/data/transforms/_transforms.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,9 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
112112

113113
return inpt
114114

115+
def transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
116+
return self._transform(inpt, params)
117+
115118

116119
@register()
117120
class ConvertPILImage(T.Transform):
@@ -133,4 +136,7 @@ def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
133136

134137
inpt = Image(inpt)
135138

136-
return inpt
139+
return inpt
140+
141+
def transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
142+
return self._transform(inpt, params)

0 commit comments

Comments
 (0)