距離変換画像(colormap)の保存方法

実現したいこと

pythonで距離変換画像を保存したいと考えています。

発生している問題・エラーメッセージ

まず、結果画像は以下のようになります。（スクリーンショットして切り取ったもの）

ひび割れの二値化画像に対して、距離変換を行いました。色については'cmap=jet'としています。

cv2.imwriteで保存したところ、全面黒の画像として保存されてしまいました。
matplotlibのimsaveで保存したところ、想定していた色とは違うもので保存されていました(下図)。

該当のソースコード

python

# 距離変換する。dist = cv2.distanceTransform(bin_img, cv2.DIST_L2, 5) #bin_imgは二値化画像 print(dist.shape) #(256, 256)plt.imshow(dist, cmap="jet") #結果表示 plt.imsave("保存先パス", dist)cv2.imwrite("保存先パス", dist)

補足情報（FW/ツールのバージョンなど）

googlecolabを使用しています。