Grad-CAM生成時,モデルのどの層を指定するべきか
解決したいこと
GradCAMの実装についてです.
モデルのどの層から特徴量の抽出を行えばよいのかが分かりません.
以下に,現在使用しているモデルのアーキテクチャをお見せします.
とても長いですが,注目すべきは最後の方の層だと思います.(メモリの関係で最初の方は省略しています)
モデルのsummary
実際にはResNet50v2 + ArcFaceのモデルを構築し,写真はその最後の方を載せています.
※ArcfaceLayerは自作レイヤーになります.
自分で試したこと
調べたところ,最後の畳み込み層(conv層)を指定するとよいとのことでしたので,最後から11層目の"conv5_block3_3_conv"をしていたのですが,その一つ手前のconv5_block3_outなどでもGradCAMは生成されるので,どの層を指定するべきなのかが分からなくなってしまいました.
ResNet50v2の場合はこのままconv5_block3_3_convを設定すればよいのでしょうか.
0