必要部分のみコピーしています。ポイントは「原画像をグレースケール⇒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