修改coco_name_lists檔案後出錯

賴韋任

您好，我只想保留"good"(有戴口罩)、"none"(口罩沒戴好)、"bad"(沒戴口罩)這三個自己新增的label，原本預設的80個label全部刪掉，於是我將"host_lib_competition\host_lib\python\common\class_lists\coco_name_lists"這個檔案修改成只有自己新增的三個label，但執行程式時（python main.py -t KL520-cam_dme_async_post_host_yolov5），會出現如右邊指令視窗的錯誤：

想請問有甚麼解決方式嗎？

Ethon Lin

Hello,

看起來可能是模型的長寬與你設定的數值不符合，導致最後在整理結果時出錯，可以透過Netron (https://netron.app/)開啟你的onnx模型，確認模型最上端的輸入尺寸是否是你設定的480x256，若不是的話請根據模型的尺寸設定model_input_row / col。

若還是有問題的話，也可以提供onnx與NEF檔讓我們協助分析。

賴韋任

尺寸確認無誤，壓縮檔為我們的onnx與NEF檔。

賴韋任

E.zip

Ethon Lin

Hello, 我這邊試著用你的NEF去跑範例，看起來是可以正常推論的，我的修改如下

1. model_file改成你的NEF
2. class_path改成你所提到的三種類別

後續直接執行後 python main.py -t KL520-cam_dme_async_post_host_yolov5 可以正常作動，我把我修改的cam_dme_async_post_host_yolov5.py與class_list提供給你參考

example.zip

另外，內文提到你是跑cam_dme_async_post_host_yolov5.py，但截圖給的是dme_async_pre_post_host_yolov5.py的參數，還請注意修改的檔案與執行的命令是否相符，若要將你的NEF模型放在dme_async_pre_post_host_yolov5.py執行，一樣是修改上述兩個參數即可，還請再試試。