heatmap-generator
Advanced tools
自动提取神经网络的中间层特征 并生成热力图 支持 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)
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
A Python package to generate heatmaps from deep learning models.
We found that heatmap-generator demonstrated a healthy version release cadence and project activity because the last version was released less than a year ago. It has 1 open source maintainer collaborating on the project.
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.
Security News
/Product
Socket Firewall Free is now bundled into Docker Hardened Images, adding build-time and dependency-install supply chain protection on top of hardened base images for Node.js, Python, and Rust.
Security News
Season’s greetings from Socket, and here’s to a calm end of year: clean dependencies, boring pipelines, no surprises.
Research
/Security News
Impostor NuGet package Tracer.Fody.NLog typosquats Tracer.Fody and its author, using homoglyph tricks, and exfiltrates Stratis wallet JSON/passwords to a Russian IP address.
