lxmwust/ag-cnn

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")

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 was 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.

AG-CNN Method

Result

Model

You can also try my models here:

Google Drive: https://drive.google.com/drive/folders/1WREzoYmwzw0gOpfobty3quySJaTDAXUi?usp=sharing

Baidu Cloud: https://pan.baidu.com/s/15Juub8f7gaBN826QuGGo8Q

Docker

You can use my docker container to avoid version issues:

https://hub.docker.com/r/ianren/cuda9.0-cudnn7.0-tf1.5-keras2.1-pytorch1.0?utm_source=docker4mac_2.2.0.0&utm_medium=repo_open&utm_campaign=referral