SSDに使用できるCNNについて

こんにちは。

こちらのドキュメンテーション(SSD 深層学習を使用したオブジェクト検出 - MATLAB & Simulink - MathWorks 日本)を参考にSSDの学習を行っています。

そこでbasenetworkとしてsqueezenetやresnet18を使用したいのですが、sddLayersではVGG16,resnet50,resnet101でなければ使用できなとエラーが返されてしまいます。この３つのCNNでないと学習できないのでしょうか。

ドキュメンテーションにはresnet18なども使用可能とあるのですが、その３つ以外のCNNを使用するにはどうすればよいのでしょうか。

よろしくお願いいたします。

Answer 1

[Kenta]

こんにちは、こちらのURLをご覧ください。

https://jp.mathworks.com/help/vision/ug/create-ssd-object-detection-network.html?lang=en

resnet18を読み込んだ後、上のドキュメントのように入力層を変更し、さらに、deepnetworkdesignerにインポートして、たとえば出力が14*14*512になるように後半をdeleteし、再度ワークスペースに読み込みます。するとほかのネットワークでもできると思います。

[HY]

ありがとうございます。

うまく作成できました。