Ien001 / AG-CNN Public

Notifications You must be signed in to change notification settings
Fork 20
Star 51

This is a reimplementation of AG-CNN. ("Thorax Disease Classification with Attention Guided Convolutional Neural Network","Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for Thorax Disease Classification")

51 stars 20 forks Branches Tags Activity

Star

Notifications

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
__pycache__		__pycache__
best_model		best_model
labels		labels
Diagnose like a Radiologist- Attention Guided Convolutional Neural Network for Thorax Disease Classification.pdf		Diagnose like a Radiologist- Attention Guided Convolutional Neural Network for Thorax Disease Classification.pdf
LICENSE		LICENSE
README.md		README.md
Screen Shot 2019-04-03 at 11.45.38 AM.png		Screen Shot 2019-04-03 at 11.45.38 AM.png
model.py		model.py
read_data.py		read_data.py
result.png		result.png
test.py		test.py
train.py		train.py

Repository files navigation

Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for Thorax Disease Classification

This is a reimplementation of paper : Diagnose like a Radiologist: Attention Guided Convolutional Neural Network for Thorax Disease Classification (AG-CNN)

Recently, the paper is accpeted in PRL 2020 with title "Thorax disease classification with attention guided convolutional neural network"

In this repo, you can train & test your own models, more details are shown in the code.