heatmap-generator

A Python package to generate heatmaps from deep learning models.

0.1.2

PyPI

Maintainers: 1

自动提取神经网络的中间层特征并生成热力图支持 GPU 加速（可选）支持 .mat 格式（用于进一步分析）支持 .png 输出（用于可视化）可直接 import 在 Python 代码或 Jupyter Notebook 里使用

pip install heatmap_generator

快速开始 from heatmap_generator import HeatmapGenerator

初始化热力图生成器（加载模型）

heatmap_gen = HeatmapGenerator(model_name="rpcanetma9",model_path="/Users/yourname/My_mission/API/RPCANet_Code/heatmap_package/heatmap_generator/result/20240519T07-24-39_rpcanetma9_nudt/best.pkl", use_cuda=True)

生成热力图，保存为 .mat 和 .png

heatmap = heatmap_gen.generate_heatmap( img_path="/Users/yourname/My_mission/API/RPCANet_Code/datasets/NUDT-SIRST/test/images/001101.png", data_name="NUDT-SIRST_test_images_001101", output_mat="/Users/yourname/My_mission/API/RPCANet_Code/heatmap_test_2.20/mat", output_png="/Users/yourname/My_mission/API/RPCANet_Code/heatmap_test_2.20/png" )

FAQs

What is heatmap-generator?

Is heatmap-generator well maintained?

Did you know?

Socket for GitHub automatically highlights issues in each pull request and monitors the health of all your open source dependencies. Discover the contents of your packages and block harmful activity before you install or update your dependencies.

Install