Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add GradCAM integration - Make YOLOv5 Interpretable #10649

Closed
wants to merge 494 commits into from
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
494 commits
Select commit Hold shift + click to select a range
771c314
.
pourmand1376 Apr 8, 2023
5e6e769
.
pourmand1376 Apr 8, 2023
cd7a30e
.
pourmand1376 Apr 8, 2023
917af3a
.
pourmand1376 Apr 8, 2023
a573cd2
.
pourmand1376 Apr 8, 2023
35297a7
.
pourmand1376 Apr 8, 2023
e76220a
.
pourmand1376 Apr 8, 2023
c096d3e
.
pourmand1376 Apr 8, 2023
76edd75
.
pourmand1376 Apr 8, 2023
70b55de
.
pourmand1376 Apr 8, 2023
d59d129
.
pourmand1376 Apr 8, 2023
2127edc
.
pourmand1376 Apr 8, 2023
3b049ff
.
pourmand1376 Apr 8, 2023
22ce832
.
pourmand1376 Apr 8, 2023
74e96ef
.
pourmand1376 Apr 8, 2023
9243f8c
.
pourmand1376 Apr 8, 2023
7615e38
.
pourmand1376 Apr 8, 2023
29dbabc
.
pourmand1376 Apr 8, 2023
f175876
.
pourmand1376 Apr 8, 2023
9a5c869
/
pourmand1376 Apr 8, 2023
fbbd11f
.
pourmand1376 Apr 8, 2023
6630bf4
.
pourmand1376 Apr 8, 2023
35b786f
.
pourmand1376 Apr 8, 2023
cf356e3
.
pourmand1376 Apr 8, 2023
3a700e0
.
pourmand1376 Apr 8, 2023
996a998
.
pourmand1376 Apr 8, 2023
6beb4ad
.
pourmand1376 Apr 8, 2023
1393393
.
pourmand1376 Apr 8, 2023
8e66ce8
.
pourmand1376 Apr 8, 2023
d1a8836
.
pourmand1376 Apr 8, 2023
6dbca54
.
pourmand1376 Apr 8, 2023
87ecb67
.
pourmand1376 Apr 8, 2023
d987e94
.
pourmand1376 Apr 8, 2023
6b2bdb6
.
pourmand1376 Apr 8, 2023
6419558
.
pourmand1376 Apr 8, 2023
ea40e77
.
pourmand1376 Apr 8, 2023
4064830
.
pourmand1376 Apr 8, 2023
4afb0fa
.
pourmand1376 Apr 8, 2023
dd13add
.
pourmand1376 Apr 8, 2023
7e02307
.
pourmand1376 Apr 8, 2023
1452cf1
.
pourmand1376 Apr 8, 2023
e2bda86
..
pourmand1376 Apr 8, 2023
b17e1a5
.
pourmand1376 Apr 8, 2023
e663e70
.
pourmand1376 Apr 8, 2023
64b22d9
.
pourmand1376 Apr 8, 2023
d7914ef
.
pourmand1376 Apr 8, 2023
d31eadb
.
pourmand1376 Apr 8, 2023
5a69e5b
.
pourmand1376 Apr 8, 2023
85aac3d
.
pourmand1376 Apr 8, 2023
29f7c89
.
pourmand1376 Apr 8, 2023
68ebe49
.
pourmand1376 Apr 8, 2023
f0ea78d
.
pourmand1376 Apr 8, 2023
7238184
.
pourmand1376 Apr 8, 2023
51dbe4a
.
pourmand1376 Apr 8, 2023
e898e4e
.
pourmand1376 Apr 8, 2023
b1aa315
.
pourmand1376 Apr 8, 2023
baef2a5
.
pourmand1376 Apr 8, 2023
0aa1316
.
pourmand1376 Apr 8, 2023
b91fcfe
.
pourmand1376 Apr 8, 2023
b4df1c9
.
pourmand1376 Apr 8, 2023
b52726b
.
pourmand1376 Apr 8, 2023
639ef30
.
pourmand1376 Apr 8, 2023
1f8c42a
.
pourmand1376 Apr 9, 2023
7ba0710
.
pourmand1376 Apr 9, 2023
cd6341c
.
pourmand1376 Apr 9, 2023
d9fe033
.
pourmand1376 Apr 9, 2023
a5c2b6b
.
pourmand1376 Apr 9, 2023
7724b76
.
pourmand1376 Apr 9, 2023
2b547c2
.
pourmand1376 Apr 9, 2023
644c8b5
.
pourmand1376 Apr 9, 2023
8a92e4b
.
pourmand1376 Apr 9, 2023
1c67526
.
pourmand1376 Apr 9, 2023
373cac2
.
pourmand1376 Apr 9, 2023
2b00c99
.
pourmand1376 Apr 9, 2023
d1c6002
.
pourmand1376 Apr 9, 2023
3b9eeed
.
pourmand1376 Apr 9, 2023
852ee3c
.
pourmand1376 Apr 9, 2023
e0d9b21
.
pourmand1376 Apr 10, 2023
69f3ce4
.
pourmand1376 Apr 10, 2023
6309000
.
pourmand1376 Apr 10, 2023
d2de181
.
pourmand1376 Apr 10, 2023
b726c53
.
pourmand1376 Apr 10, 2023
0782010
.
pourmand1376 Apr 10, 2023
d53773b
.
pourmand1376 Apr 10, 2023
38a55a1
.
pourmand1376 Apr 10, 2023
d876d1c
.
pourmand1376 Apr 10, 2023
c588560
.
pourmand1376 Apr 10, 2023
f1cb3c0
.
pourmand1376 Apr 10, 2023
beb2bef
.
pourmand1376 Apr 10, 2023
f628117
.
pourmand1376 Apr 10, 2023
65fb0a6
.
pourmand1376 Apr 10, 2023
90a4ad5
.
pourmand1376 Apr 10, 2023
ca7ab5f
.
pourmand1376 Apr 10, 2023
ca0a4c0
.
pourmand1376 Apr 10, 2023
8470022
.
pourmand1376 Apr 10, 2023
abf7ca8
.
pourmand1376 Apr 10, 2023
f2ac46d
.
pourmand1376 Apr 10, 2023
8755e7d
.
pourmand1376 Apr 10, 2023
ff136db
.
pourmand1376 Apr 10, 2023
9845f87
.
pourmand1376 Apr 10, 2023
fcb2df7
.
pourmand1376 Apr 10, 2023
fe0eda0
.
pourmand1376 Apr 10, 2023
26a2e71
.
pourmand1376 Apr 10, 2023
66cb90e
.
pourmand1376 Apr 10, 2023
ff6d212
.
pourmand1376 Apr 10, 2023
7026a62
.
pourmand1376 Apr 10, 2023
fcd8fae
.
pourmand1376 Apr 10, 2023
b91da71
.
pourmand1376 Apr 10, 2023
2f16624
.
pourmand1376 Apr 10, 2023
09a5cbb
.
pourmand1376 Apr 10, 2023
ce7af87
.
pourmand1376 Apr 10, 2023
2813ffe
.
pourmand1376 Apr 10, 2023
89416fb
.
pourmand1376 Apr 10, 2023
ad2034d
.
pourmand1376 Apr 10, 2023
f1db726
.
pourmand1376 Apr 10, 2023
23b6826
.
pourmand1376 Apr 10, 2023
8c5e5fe
.
pourmand1376 Apr 10, 2023
47fef8a
.
pourmand1376 Apr 10, 2023
526f18a
.
pourmand1376 Apr 10, 2023
f75b8cc
.
pourmand1376 Apr 10, 2023
650ea20
.
pourmand1376 Apr 11, 2023
b863473
.
pourmand1376 Apr 11, 2023
1457e10
.
pourmand1376 Apr 11, 2023
573dfce
.
pourmand1376 Apr 11, 2023
c80e3d0
.
pourmand1376 Apr 11, 2023
e1add2f
.
pourmand1376 Apr 11, 2023
a184075
.
pourmand1376 Apr 11, 2023
47d27f1
.
pourmand1376 Apr 11, 2023
88245ef
.
pourmand1376 Apr 11, 2023
1215fea
.
pourmand1376 Apr 11, 2023
77b9c1c
.
pourmand1376 Apr 11, 2023
93a1af5
.
pourmand1376 Apr 11, 2023
4ae01a5
.
pourmand1376 Apr 11, 2023
465eb4f
.
pourmand1376 Apr 11, 2023
9ff88cc
.
pourmand1376 Apr 11, 2023
0fa5e18
.
pourmand1376 Apr 11, 2023
bca37a8
.
pourmand1376 Apr 11, 2023
06d2009
.
pourmand1376 Apr 11, 2023
2705c87
.
pourmand1376 Apr 11, 2023
44c01b0
.
pourmand1376 Apr 11, 2023
d9322a8
.
pourmand1376 Apr 11, 2023
41400ad
.
pourmand1376 Apr 11, 2023
e032be9
.
pourmand1376 Apr 11, 2023
16e507d
.
pourmand1376 Apr 11, 2023
77df050
.
pourmand1376 Apr 11, 2023
1888bca
.
pourmand1376 Apr 11, 2023
cc5b7ac
.
pourmand1376 Apr 11, 2023
4ba10a2
.
pourmand1376 Apr 11, 2023
7283fe0
.
pourmand1376 Apr 11, 2023
51b2345
.
pourmand1376 Apr 11, 2023
ac6639e
.
pourmand1376 Apr 11, 2023
68acff1
.
pourmand1376 Apr 11, 2023
9b212cb
.
pourmand1376 Apr 11, 2023
21d3d81
.
pourmand1376 Apr 11, 2023
9dee756
.
pourmand1376 Apr 11, 2023
d294826
.
pourmand1376 Apr 11, 2023
d21d55b
.
pourmand1376 Apr 11, 2023
a53708c
.
pourmand1376 Apr 11, 2023
3fc9ee3
.
pourmand1376 Apr 11, 2023
3a53b02
.
pourmand1376 Apr 11, 2023
5093ab6
.
pourmand1376 Apr 11, 2023
30d1059
.
pourmand1376 Apr 11, 2023
c5ec3fc
Merge branch 'master' into add_gradcam
pourmand1376 Apr 11, 2023
4758453
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 11, 2023
676fef8
Merge branch 'add_gradcam_test' into add_gradcam
pourmand1376 Apr 11, 2023
28f3ef6
Merge branch 'add_gradcam' of github.com:pourmand1376/yolov5 into add…
pourmand1376 Apr 11, 2023
bcb93b4
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 11, 2023
e8d02bb
.
pourmand1376 Apr 11, 2023
9ecf49a
.
pourmand1376 Apr 11, 2023
2e04f35
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 11, 2023
fd18b86
add mean!
pourmand1376 Apr 11, 2023
29bfc6d
.
pourmand1376 Apr 11, 2023
b5fd9a2
.
pourmand1376 Apr 11, 2023
552e001
.
pourmand1376 Apr 11, 2023
d00377d
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 17, 2023
83a3c09
.
pourmand1376 Apr 19, 2023
ea80eff
.
pourmand1376 Apr 19, 2023
59b84d5
Merge branch 'add_gradcam_test' into add_gradcam
pourmand1376 Apr 19, 2023
bfd9414
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 19, 2023
34f0600
Merge branch 'master' into add_gradcam
pourmand1376 Apr 19, 2023
556548e
update file
pourmand1376 Apr 19, 2023
026660a
remove file
pourmand1376 Apr 19, 2023
d8fb55d
Merge branch 'add_Gradcam_test' into add_gradcam
pourmand1376 Apr 19, 2023
bbf23c7
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Apr 19, 2023
7b1923a
remove detect.py usuage
pourmand1376 May 1, 2023
4b15b24
Merge branch 'master' into add_gradcam
pourmand1376 May 1, 2023
90d8187
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 1, 2023
d3e6827
add changes
pourmand1376 May 1, 2023
fb46e5b
Merge branch 'add_gradcam' of github.com:pourmand1376/yolov5 into add…
pourmand1376 May 1, 2023
e8f8e96
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 1, 2023
f99233a
remove iou_thres
pourmand1376 May 1, 2023
3371246
Merge branch 'add_gradcam' of github.com:pourmand1376/yolov5 into add…
pourmand1376 May 1, 2023
692240c
fix function
pourmand1376 May 1, 2023
581f457
add output
pourmand1376 May 1, 2023
d48d339
exclude ipynb
pourmand1376 May 1, 2023
5307867
Merge branch 'master' into add_gradcam
pourmand1376 May 6, 2023
a6d688f
Merge branch 'master' into add_gradcam
pourmand1376 May 16, 2023
aeabfc5
add path for interpretability
pourmand1376 May 26, 2023
3d90dc6
add cpu
pourmand1376 May 27, 2023
c84dae1
add logger
pourmand1376 May 27, 2023
134c2c4
add default method
pourmand1376 May 27, 2023
7ed7e02
.
pourmand1376 May 27, 2023
91f8967
.
pourmand1376 May 27, 2023
65f6c98
fix stride
pourmand1376 May 27, 2023
0740ec7
.
pourmand1376 May 27, 2023
145377c
add inheritance
pourmand1376 May 27, 2023
0b5daf6
.
pourmand1376 May 27, 2023
a05033a
add yolo
pourmand1376 May 27, 2023
955e695
add super()
pourmand1376 May 27, 2023
1e8d409
remove model
pourmand1376 May 27, 2023
403222a
add type check
pourmand1376 May 27, 2023
58b8e21
.
pourmand1376 May 27, 2023
3b5afcd
.
pourmand1376 May 27, 2023
28249f1
.
pourmand1376 May 27, 2023
71d8ed0
add topk
pourmand1376 May 27, 2023
e0afe62
fix
pourmand1376 May 27, 2023
2871061
.
pourmand1376 May 27, 2023
cc12c97
.
pourmand1376 May 27, 2023
cbed483
keep only top k percent
pourmand1376 May 27, 2023
855f17b
add default
pourmand1376 May 27, 2023
dab31b1
add crop
pourmand1376 May 27, 2023
e3c60a3
add crop
pourmand1376 May 27, 2023
d8cea82
add breakpoint
pourmand1376 May 27, 2023
ac6c919
add max
pourmand1376 May 27, 2023
777b467
add 255
pourmand1376 May 27, 2023
da4dfa8
.
pourmand1376 May 27, 2023
d0f3a8e
to device
pourmand1376 May 27, 2023
b5dbe12
add cuda
pourmand1376 May 27, 2023
012c363
fix cuda
pourmand1376 May 27, 2023
ee8feb3
.
pourmand1376 May 27, 2023
5a895aa
add .
pourmand1376 May 27, 2023
148ce91
add device
pourmand1376 May 27, 2023
8c026cb
.
pourmand1376 May 27, 2023
2f9622c
add error
pourmand1376 May 27, 2023
dadad8f
.
pourmand1376 May 27, 2023
1a29ca1
.
pourmand1376 May 27, 2023
87ee4bb
add error
pourmand1376 May 27, 2023
91b1753
add logger
pourmand1376 May 27, 2023
22ac434
.
pourmand1376 May 27, 2023
4831e0b
.
pourmand1376 May 27, 2023
78f3d1a
.
pourmand1376 May 27, 2023
c36f097
add makefile
pourmand1376 May 29, 2023
b4561c6
.
pourmand1376 May 29, 2023
a175b61
add old_target_method
pourmand1376 May 29, 2023
0abce09
add
pourmand1376 May 29, 2023
462bdeb
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] May 31, 2023
88ea3ef
add image
pourmand1376 Jun 16, 2023
80e88f9
[pre-commit.ci] auto fixes from pre-commit.com hooks
pre-commit-ci[bot] Jun 16, 2023
b99218a
add gradcam++
pourmand1376 Jun 16, 2023
2127d85
Merge branch 'add_gradcam' of github.com:pourmand1376/yolov5 into add…
pourmand1376 Jun 16, 2023
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions .pre-commit-config.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -62,6 +62,7 @@ repos:
- id: codespell
args:
- --ignore-words-list=crate,nd,strack,dota
- --exclude-file=explainer/demo.ipynb

#- repo: https://github.com/asottile/yesqa
# rev: v1.4.0
Expand Down
9 changes: 9 additions & 0 deletions Makefile
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
run_interpretability:
python explainer/explainer.py --weights runs/train/exp8/weights/best.pt \
--source /mnt/new_drive/pourmand/KUMC-Harvard/KUMC/train2019/images/ \
--method GradCAM --keep-only-topk 1 --crop True --device 0

run_interpretability_old:
python explainer/explainer.py --weights runs/train/exp8/weights/best.pt \
--source /mnt/new_drive/pourmand/KUMC-Harvard/KUMC/train2019/images/ \
--method GradCAM --keep-only-topk 1 --crop True --device 0 --use-old-target-method True
634 changes: 634 additions & 0 deletions explainer/demo.ipynb

Large diffs are not rendered by default.

Loading