Skip to content

[Feat] Add conversion to onnx

Mohamed Mssaouri requested to merge feat/AddConversionToOnnx into main

Goal

The goal of this MR is to add a python plugin that performs conversion from yolo-darknet and crcnn-mmdet to onnx

To perform a conversion to onnx, follow the steps below:

1 - cd VIAME-build/install

2- bin/kwiver runner configs/pipelines/convert_to_onnx.pipe -s "onnx_convert:model_path =" /path/to/yolo.weights/or/trained_detector.zip -s "onnx_convert:onnx_model_prefix =" /path//prefix/of/the/generated/onnx"

Edited by Mohamed Mssaouri

Merge request reports