PROCESSOR-SDK-AM62A: Is there any script for onnx2prototxt

Ryuuichi machida

Hello,

I'm looking for onnx2prototxt for TIDL environment.
I understand that you have release script for ssd, retinanet, yolo model to convert pth to onnx and prototxt.
However what I'm looking for is script for generating prototxt from onnx.
(Like this https://github.com/axinc-ai/export-to-onnx/blob/master/onnx2prototxt.py)

Do you have any repo information ?

BR,

over 1 year ago

+1 Reese Grimsley over 1 year ago

TI__Genius 16936 points

Hello Machida-san,

I am not aware of a generalized function of this purpose across all our model-training repositories. We have targeted specific architectures for the detection head. There are some functions specific to these architectures:

yolov5: https://github.com/TexasInstruments/edgeai-yolov5/blob/f45afc762fc4cb780c614e77ca17e96061fafb49/export.py#L89
yolox (would require modification to only use ONNX): https://github.com/TexasInstruments/edgeai-yolox/blob/cbdd9d85def4e55e3fbbf38c155e183e90ed5cc6/tools/export_onnx.py#L117

Full doc on meta-architectures and prototxt's for object detection networks: https://github.com/TexasInstruments/edgeai-tidl-tools/blob/master/docs/tidl_fsg_od_meta_arch.md

0 Ryuuichi machida over 1 year ago in reply to Reese Grimsley

Guru 12980 points

Hello Reese-san,

Thank you for your reply.
Understood.

BR,

Processors

Processors forum

PROCESSOR-SDK-AM62A: Is there any script for onnx2prototxt