Skip to content

Commit db32e48

Browse files
committed
Change swap_rgb to False
1 parent 43bfb38 commit db32e48

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/otx/algo/detection/rtmdet.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -169,7 +169,7 @@ def _exporter(self) -> OTXModelExporter:
169169
std=self.std,
170170
resize_mode="fit_to_window_letterbox",
171171
pad_value=114,
172-
swap_rgb=True,
172+
swap_rgb=False, # TODO (sungchul): revert to True after merging releases/2.0.0 to develop
173173
via_onnx=True,
174174
onnx_export_configuration={
175175
"input_names": ["image"],
@@ -179,7 +179,6 @@ def _exporter(self) -> OTXModelExporter:
179179
"boxes": {0: "batch", 1: "num_dets"},
180180
"labels": {0: "batch", 1: "num_dets"},
181181
},
182-
"opset_version": 11,
183182
"autograd_inlining": False,
184183
},
185184
output_names=["bboxes", "labels", "feature_vector", "saliency_map"] if self.explain_mode else None,

0 commit comments

Comments
 (0)