深層学習による画像の回帰でのgradcam

深層学習によるX線画像の回帰を行っております。

https://jp.mathworks.com/help/deeplearning/ref/gradcam.html#mw_7d8ef797-7a5e-444c-8302-16fe0c3caa1d

上記ページの通りにgradcamを用いているのですが、クラス分類のgradcamのように色を薄くして原画像も見えるようにしたいです。

ご教示ください。

Answer 1

[Atsushi Ueno]

必要部分のみコピーしています。ポイントは「原画像をグレースケール⇒RGBにする」(そうしないとjetカラーマップに乗ってしまう)点です。

load digitsRegressionNet
[XTest,~,YTest] = digitTest4DArrayData;
testDigit = XTest(:,:,:,501); % 原画像
scoreMap = gradCAM(net,testDigit,@(x)x,'ReductionLayer','fc','FeatureLayer','relu_4');

ax(1) = subplot(1,2,1);
imshow(testDigit);
colormap(ax(1),'gray')

ax(2) = subplot(1,2,2);
imshow(repmat(testDigit,1,1,3)); % 原画像をグレースケール⇒RGBにする
hold on
imagesc(rescale(scoreMap),'AlphaData',0.5); % クラス分類のgradcamに倣い透明度を50%にする
colormap(ax(2),'jet')
hold off