Aaron Huang

custom yolov4-tiny can inference but no output on yolov5-ncnn-android

8

大神好，我想用自己訓練的yolov4-tiny 在 android-ncnn 檢測目標 以下是我的yolov4-tiny-opt.param  這是我的模型訓練過程 我用WongKinYiu的yolov4-csp訓練了六個類別的yolov4-tiny 以下是程式碼的網址 https://github.com/WongKinYiu/ScaledYOLOv4/tree/yolov4-csp 參考這個issue WongKinYiu/ScaledYOLOv4#41 以及更改cfg檔(classes=6 ＆ yolo層輸出的filter=33) 成功訓練出pytorch版本的yolov4-tiny (6-category） 並用ncnn中的darknet2ncnn與ncnnoptimize轉換成ncnn的模型與優化 使用您的android_YOLOV5_NCNN實現推論 但在執行下面這行 ex.extract("output", blob); 一直都沒有輸出結果，想請教大神我可能是哪裡出了問題