File tree Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Expand file tree Collapse file tree 1 file changed +10
-0
lines changed Original file line number Diff line number Diff 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
You can’t perform that action at this time.
0 commit comments