xw-hu/AGNet

C++MIT

AGNet: Attention-Guided Network for Surgical Tool Presence Detection

by Xiaowei Hu, Lequan Yu, Hao Chen, Jing Qin and Pheng-Ann Heng

This implementation is written by Xiaowei Hu at the Chinese University of Hong Kong.

Citation

@incollection{hu2017agnet,   
  title={{AGNet}: Attention-Guided Network for Surgical Tool Presence Detection},   
  author={Hu, Xiaowei and Yu, Lequan and Chen, Hao and Qin, Jing and Heng, Pheng-Ann},   
  booktitle={Deep Learning in Medical Image Analysis and Multimodal Learning for Clinical Decision Support},   
  pages={186--194},   
  year={2017}         
}

Installation

Clone the AGNet repository, and we'll call the directory that you cloned AGNet into AGNet.
```
git clone https://github.com/xw-hu/AGNet.git
```
Build Caffe

*This model is tested on Ubuntu 16.04, CUDA 8.0, cuDNN 5.0

Follow the Caffe installation instructions here: http://caffe.berkeleyvision.org/installation.html
```
make all -j XX
make matcaffe
```
Download data in AGNet/data/.

Train and Test

Download the pre-trained model in AGNet/models/.
Run AGNet/matlab/AGNet_train.m or AGNet/matlab/AGNet_test.m.