Conversion from frozen graph to ONNX prediction mismatch

thtrieu / darkflow

Translate darknet to tensorflow. Load trained weights, retrain/fine-tune using tensorflow, export constant graph def to mobile devices

GNU General Public License v3.0

6.13k stars 2.08k forks source link

Open VinuthaRaghavendra opened 4 years ago

VinuthaRaghavendra commented 4 years ago

Conversion of frozen graph to onnx has prediction mismatch (used tf2Onnx to convert) always has prediction score equal to 90.

But able to convert frozen graph from the tiny-yolo-voc.weights to .onnx with the same command and predictions are accurate.

Used winmltools to check inference