Hover-Trans: Anatomy-aware HoVer-Transformer for ROI-free Breast Cancer Diagnosis in Ultrasound Images

Introduction

The implementation of:
Hover-Trans: Anatomy-aware HoVer-Transformer for ROI-free Breast Cancer Diagnosis in Ultrasound Images

Requirements

• python 3.9
• Pytorch 1.10.1
• torchvision 0.11.2
• opencv-python
• pandas
• scipy

Setup

Installation

Clone the repo and install required packages:

git clone https://github.com/yuhaomo/HoVerTrans.git
cd HoVerTrans
pip install -r requirements.txt

Dataset

• You can download our dataset (GDPH&SYSUCC) and unpack them into the ./data folder.

./data
└─GDPH&SYSUCC
      ├─label.csv
      └─img
          ├─benign(0).png
          ├─benign(1).png
          ├─benign(2).png
          ├─malignant(0).png
          ├─malignant(1).png
          ...

• The format of the label.csv is as follows:

+------------------+-------+
| name             | label |
+------------------+-------+
| benign(0).png    |   0   |
| benign(1).png    |   0   |
| benign(2).png    |   0   |
| malignant(0).png |   1   |
| malignant(1).png |   1   |
...

Training

python train.py --data_path ./data/GDPH&SYSUCC/img --csv_path ./data/GDPH&SYSUCC/label.csv --batch_size 32 --class_num 2 --epochs 250 --lr 0.0001 

Citation

If you find this repository useful or use our dataset, please consider citing our work:

@ARTICLE{10015121,
  author={Mo, Yuhao and Han, Chu and Liu, Yu and Liu, Min and Shi, Zhenwei and Lin, Jiatai and Zhao, Bingchao and Huang, Chunwang and Qiu, Bingjiang and Cui, Yanfen and Wu, Lei and Pan, Xipeng and Xu, Zeyan and Huang, Xiaomei and Li, Zhenhui and Liu, Zaiyi and Wang, Ying and Liang, Changhong},
  journal={IEEE Transactions on Medical Imaging}, 
  title={HoVer-Trans: Anatomy-aware HoVer-Transformer for ROI-free Breast Cancer Diagnosis in Ultrasound Images}, 
  year={2023},
  volume={},
  number={},
  pages={1-1},
  doi={10.1109/TMI.2023.3236011}}