chris/frigate
1
0
Fork 0
forked from Github/frigate
Code Pull Requests Packages Activity

upgrade to latest openvino version (#11563)

Browse Source
This commit is contained in:
Blake Blackshear
2024-05-27 15:49:35 -05:00
committed by GitHub
parent a86e22e0fc
commit bfeb7b8a96
6 changed files with 83 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
docker/main/build_ov_model.py Normal file
View File

@@ -0,0 +1,11 @@
import openvino as ov
from openvino.tools import mo
ov_model = mo.convert_model(
"/models/ssdlite_mobilenet_v2_coco_2018_05_09/frozen_inference_graph.pb",
compress_to_fp16=True,
transformations_config="/usr/local/lib/python3.9/dist-packages/openvino/tools/mo/front/tf/ssd_v2_support.json",
tensorflow_object_detection_api_pipeline_config="/models/ssdlite_mobilenet_v2_coco_2018_05_09/pipeline.config",
reverse_input_channels=True,
)
ov.save_model(ov_model, "/models/ssdlite_mobilenet_v2.xml")