Yolo Heatmaps

A utility for generating heatmaps of YOLOv5 using Layerwise Relevance Propagation (LRP/CRP).
Alternatives To Yolo Heatmaps
Project NameStarsDownloadsRepos Using ThisPackages Using ThisMost Recent CommitTotal ReleasesLatest ReleaseOpen IssuesLicenseLanguage
2 months agon,ullotherC
YOLOv4 / Scaled-YOLOv4 / YOLO - Neural Networks for Object Detection (Windows and Linux version of Darknet )
Cvpr2023 Paper Code Interpretation11,221
8 hours ago38
cvpr2022/cvpr2021/cvpr2020/cvpr2019/cvpr2018/cvpr2017 论文/代码/解读/直播合集,极市团队整理
Cvpr2023 Papers With Code10,861
7 days ago5
CVPR 2023 论文和开源项目合集
5 hours ago2September 08, 2022475mitTypeScript
Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.
Computervision Recipes8,817
2 months ago65mitJupyter Notebook
Best Practices, code samples, and documentation for Computer Vision.
4 months ago75mitC#
Experience, Learn and Code the latest breakthrough innovations with Microsoft AI
Pytorch Grad Cam6,939120 days ago25May 20, 202262mitPython
Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.
Awesome Object Detection6,707
a year ago6
Awesome Object Detection based on handong1587 github: https://handong1587.github.io/deep_learning/2015/10/09/object-detection.html
Jetson Inference6,243
a day ago938mitC++
Hello AI World guide to deploying deep-learning inference networks and deep vision primitives with TensorRT and NVIDIA Jetson.
3 hours ago222apache-2.0Python
AutoGluon: AutoML for Image, Text, Time Series, and Tabular Data
Alternatives To Yolo Heatmaps
Select To Compare

Alternative Project Comparisons

YOLOv5 Heatmaps


A utility for generating heatmaps of YOLOv5 ultralytics/yolov5 using Layerwise Relevance Propagation (LRP/CRP). Pytorch implementation based on: moboehle/Pytorch-LRP


git clone https://github.com/akarasman/yolo-heatmaps/
cd yolo-heatmaps
pip install -r requirements.txt

CLI Use Example

python3 explain.py --source=data/images/so-and-so.jpg --weights=yolov5s.pt --explain-class='person'

Run results saved to runs/explain/exp(# of run)


  # explain.py is built on detect.py module from YOLOv5, lrp options are :
  --power POWER         Power exponent applied to weights and inputs
  --contrastive         Use contrastive relevance (CRP)
  --b1 B1               Visualization parameter for CRP - multiplier of primal part
  --b2 B2               Visualization parameter for CRP - multiplier of dual part
  --explain-class EXPLAIN_CLASS
                        Class to explain
  --conf                Confidence threshold on object
  --max-class-only      Max class only
  --box-xywh BOX_XYWH [BOX_XYWH ...]
                        Box to restrict investigation (X,Y,W,H format)
  --smooth-ks SMOOTH_KS
                        Box to restrict investigation (X,Y,W,H format)
  --box-xyxy BOX_XYXY [BOX_XYXY ...]
                        Box to restrict investigation (X,Y,X,Y format)
  --cmap CMAP           Explanation color map (default set to seismic/magma when contrastive / non-contrastive

Current version only supports YOLOv5s-x models.

Please cite our paper if you plan on using code from this repository for your work

author = {Karasmanoglou, Apostolos and Antonakakis, Marios and Zervakis, Michalis},
year = {2022},
month = {06},
pages = {1-6},
title = {Heatmap-based Explanation of YOLOv5 Object Detection with Layer-wise Relevance Propagation},
doi = {10.1109/IST55454.2022.9827744}
Popular Computer Vision Projects
Popular Object Detection Projects
Popular Machine Learning Categories
Related Searches

Get A Weekly Email With Trending Projects For These Categories
No Spam. Unsubscribe easily at any time.
Computer Vision
Object Detection