TFNet

[ACPR' 21] TFNet: Transformer Fusion Network for Ultrasound Image Segmentation.

Tao Wang, Zhihui Lai and Heng Kong

Installation

Please refer to requirements.txt

Dataset

• Download datasets: BUSI, DDTI.

• You can found Processed DDTI dataset in this link.

• After downloading the datasets, please change the dataset root path in configs/_base_/datasets.

Results

Model	Dataset	Backbone	Iterations	Batch Size	DsC	Config File
TFNet	BUSI	ResNet-50	40000	16	77.3±5.5	configs/tfnet-r50_256_40k_busi.py
TFNet	DDTI	ResNet-50	40000	16	84.6±1.7	configs/tfnet-r50_256_40k_ddti.py
TFNet	Our Dataset	ResNet-50	40000	16	87.9±1.4	configs/tfnet-r50_256_40k_breast.py

Visualization

Reference

• https://github.com/open-mmlab/mmsegmentation

Citation

@inproceedings{wang2022tfnet,
  title={TFNet: Transformer Fusion Network for Ultrasound Image Segmentation},
  author={Wang, Tao and Lai, Zhihui and Kong, Heng},
  booktitle={Asian Conference on Pattern Recognition},
  pages={314--325},
  year={2022},
  organization={Springer}
}