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.

@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}
}

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

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

3. Download data in AGNet/data/.

1. Download the pre-trained model in AGNet/models/.

2. Run AGNet/matlab/AGNet_train.m or AGNet/matlab/AGNet_test.m.