YOLO-NASのonnxでの精度が低い問題

[Hoku113]

super-gradientsで60～80%で分析で来ていたのものが、onnxで分析すると10%以下の精度で検出されてしまう問題が発生

原因

• pth $\rightarrow$ onnx に変換したときにおこる
• 現在制作しているプログラムが純粋に間違っている

このどちらかに問題がありそう

[Hoku113]

ここら辺が参考になりそう？

model.eval()
model.prep_model_for_conversion(input_size=[1, 3, 640, 640])

# Create dummy_input

# Convert model to onnx
torch.onnx.export(model, dummy_input,  "yolo_nas_m.onnx")

[Hoku113]

上記の対策及びその他の対策を実行しても特に精度が上がることはなかった。 ひとまず次にTensorRTを試してモデルの精度を確かめてみる