Skip to content

Commit 97a2323

Browse files
committed
Add rt info for ModelAPI (open-edge-platform/model_api#71)
1 parent a2bb42d commit 97a2323

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

ultralytics/yolo/engine/exporter.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -362,6 +362,16 @@ def export_openvino(self, prefix=colorstr('OpenVINO:')):
362362
model_name=self.pretty_name,
363363
framework='onnx',
364364
compress_to_fp16=self.args.half) # export
365+
ov_model.set_rt_info("YoloV8", ["model_info", "model_type"])
366+
ov_model.set_rt_info("fit_to_window_letterbox", ["model_info", "resize_type"])
367+
ov_model.set_rt_info(True, ["model_info", "reverse_input_channels"])
368+
ov_model.set_rt_info(114, ["model_info", "pad_value"])
369+
ov_model.set_rt_info([255.0], ["model_info", "scale_values"])
370+
ov_model.set_rt_info(self.args.iou, ["model_info", "iou_threshold"])
371+
labels = []
372+
for i in range(len(self.model.names)):
373+
labels.append(self.model.names[i].replace(" ", "_"))
374+
ov_model.set_rt_info(labels, ["model_info", "labels"])
365375
ov.serialize(ov_model, f_ov) # save
366376
yaml_save(Path(f) / 'metadata.yaml', self.metadata) # add metadata.yaml
367377
return f, None

0 commit comments

Comments
 (0)