Open Hoku113 opened 1 year ago
ここら辺が参考になりそう?
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")
上記の対策及びその他の対策を実行しても特に精度が上がることはなかった。 ひとまず次にTensorRTを試してモデルの精度を確かめてみる
super-gradients
で60~80%で分析で来ていたのものが、onnx
で分析すると10%以下の精度で検出されてしまう問題が発生原因
pth
$\rightarrow$onnx
に変換したときにおこるこのどちらかに問題がありそう