Skip to content

Commit 1702913

Browse files
committed
Update exporter arguments
1 parent 4aa1125 commit 1702913

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

src/otx/algo/detection/rtmdet.py

+3-4
Original file line numberDiff line numberDiff line change
@@ -171,21 +171,20 @@ def _exporter(self) -> OTXModelExporter:
171171
std=self.std,
172172
resize_mode="fit_to_window_letterbox",
173173
pad_value=114,
174-
swap_rgb=False,
174+
swap_rgb=True,
175175
via_onnx=True,
176176
onnx_export_configuration={
177177
"input_names": ["image"],
178-
"output_names": ["boxes", "labels", "masks"],
178+
"output_names": ["boxes", "labels"],
179179
"dynamic_axes": {
180180
"image": {0: "batch", 2: "height", 3: "width"},
181181
"boxes": {0: "batch", 1: "num_dets"},
182182
"labels": {0: "batch", 1: "num_dets"},
183-
"masks": {0: "batch", 1: "num_dets", 2: "height", 3: "width"},
184183
},
185184
"opset_version": 11,
186185
"autograd_inlining": False,
187186
},
188-
output_names=["bboxes", "labels", "masks", "feature_vector", "saliency_map"] if self.explain_mode else None,
187+
output_names=["bboxes", "labels", "feature_vector", "saliency_map"] if self.explain_mode else None,
189188
)
190189

191190
def forward_for_tracing(

0 commit comments

Comments
 (0)