RGB_CNN 模型的使用規則

Technical Discussion AMB82-mini
1

各位好，

我想請教關於 AMB82 是否支援自訓練的 CNN 分類模型的問題。如果可以使用，是否有特定的模型格式或類型需要遵循？

我在官網上的文件中看到，AMB82 支援 CNN 模型的描述（官網說明），但同時也在論壇中發現提到 AMB82 不支援 CNN 模型的說法（論壇討論）。這讓我對 AMB82 的 CNN 支援能力感到困惑。

此外，我嘗試使用 MobileNetV3 訓練了一個分類模型，並透過官網推薦的工具進行模型轉換，但最終轉換失敗。想請教：

  1. 是否有推薦的 CNN 模型結構（如 MobileNetV1/V2 等）可以穩定轉換並部署在 AMB82 上？
  2. 模型的結構是否有限制，是否有細節的說明文件可以參考。
    application.zip (2.9 KB)

3.使用.onnx 模型轉換，是否有需要特別注意的事項或限制？
希望能得到大家的指導與建議，謝謝！

4.附件是上傳模型後，官方回傳的error資訊

@Kelvin_Huang
@pammyleong
@KevinKL

2

您好 @josh，

AMB82 不支持 HardSwish op，因此 Mobilenetv3 結構的模型無法在AMB82 上使用。您可以嘗試 Mobilenetv2。

您可以導到此鏈接 （https://www.amebaiot.com/en/amebapro2-ai-convert-model/） 查看在訓練模型以及轉換時要注意的地方。

3

Model Training:
https://www.kaggle.com/code/rkuo2000/garbage-cnn

4

謝謝，換成MobileNet V2後就成功了

5

我也是AMB82的使用者，想詢問一下mobilenetV2最後是怎麼轉onnx，我是使用pytorch mobilenetv2最後跑出一些log，以下是轉的程式以及log的部分區域，想請問這樣是否會有問題

image
image669×421 20.2 KB

LOG:
…
IMGCLASSIFICATION_SD tick[28]
IMGCLASSIFICATION_SD tick[28]
IMGCLASSIFICATION_SD tick[29]
IMGCLASSIFICATION_SD FPS = 8.37
IMGCLASSIFICATION_SD tick[29]
IMGCLASSIFICATION_SD tick[29]
…

6

Hi @Kai-Hao ,

想請問您是使用online conversion tool 轉換模型嗎？

因爲online conversion tool 目前還沒支援轉換 mobilenetv2。您可以考慮 offline conversion tool

您可以嘗試最新的 pre release version, 我們有更新 image classification mobilenetv2 範例

7

謝謝，這幾天我再試試看，感謝你的回覆～