search

ArmDeveloperEcosystem / ml-object-detection-examples-for-imx93

ML Object Detection Examples for i.MX 93
MIT License
1 stars 3 forks source link

IndexError: list index out of range #2

Closed LiuHanHsiang closed 5 months ago

LiuHanHsiang commented 5 months ago

I'm using TrafficCamNet ONNX Model from Nvidia NGC

I convert the onnx into tflite model using model_conversion.ipynb

when I try to run python main.py resnet18_trafficcamnet_pruned.tflite it happened

////////////////////////////////// pad 0.0 convert 15.628576278686523 predict 46.76985740661621 post process 0.0 nms 0.0 nms 0.0 nms 0.0 Traceback (most recent call last): File ".\main.py", line 67, in max_output_sizes[label], IndexError: list index out of range /////////////////////////////////

.tflite file screenshot of the model in Netron.app resnet18_trafficcamnet_pruned tflite

LiuHanHsiang commented 5 months ago

ok i figured it out by changing main.py colors max_output_sizes iou_thresholds score_thresholds with 4 indexs

sandeepmistry commented 5 months ago

Hi @LiuHanHsiang,

Excellent, thanks for the update, great to hear you have this working!