MHA-UNet

KO | EN

gitlite — search

#javascript #python #unity #python3 #golang #pytorch #tensorflow #machine-learning #csharp #c #php #go

★ 36 Open GitHub ↗

[BIBM 2025] The official code for "Only Positive Cases: 5-fold High-order Attention Interaction Model for Skin Segmentation Derived Classification".

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

# Only Positive Cases: 5-fold High-order Attention Interaction Model for Skin Segmentation Derived Classification
 
 Renkai Wu, Junwei Liu, Yinghao Liu, Pengchen Liang, Guochen Ning* and Qing Chang* 
 [IEEE BIBM 25](https://doi.org/10.1109/BIBM66473.2025.11356752)

</div>

https://github.com/wurenkai/MHA-UNet/assets/124028634/921049af-4797-4db6-b6bf-b727df82ba56

**0. Main Environments.**
- python 3.8
- pytorch 1.12.0

**1. Prepare the Dataset and Pretrained weights.**

*A.Dataset* 
1- Download the ISIC 2017 train dataset from [this](https://challenge.isic-archive.com/data) link and extract both training dataset and ground truth folders inside the `/data/dataset_isic17/`. 
2- Run `Prepare_ISIC2017.py` for data preparation and dividing data to train,validation and test sets.

**Notice:**
For training and evaluating on ISIC 2018, pH2, NormalSkin and Kaggle95 follow the bellow steps: :
1- Download the ISIC 2018 train dataset from [this](https://challenge.isic-archive.com/data) link and extract both training dataset and ground truth folders inside the `/data/dataset_isic18/`. then Run ` Prepare_ISIC2018.py` for data preperation and dividing data to train,validation and test sets. 
2- Download the pH2 dataset from [this](https://www.dropbox.com/s/k88qukc20ljnbuo/PH2Dataset.rar) link and extract it then Run ` Prepare_PH2_test.py` for data preperation and dividing data to train,validation and test sets. 
3- Download the NormalSkin dataset from [this](https://universe.roboflow.com/janitha-prathapa/normalskin) link. 
4- Download the Kaggle95 dataset from [this](https://www.kaggle.com/datasets/ahdasdwdasd/our-normal-skin/data) link. 
5- The NormalSkin dataset and the Kaggle95 dataset were used as negative samples to test the model classification ability. For preparing these negative test samples, the data can be processed in the following way: 
#0 The negative dataset is without segmentation labels. It is possible to generate the same number of all-black labels as the original image in the following way: 
```
python generate_black.py
```
It should be noted that the number of generated images in the generate_black.py file needs to be modified.

#1 When the same number of all-black labels are available, the following command is executed to generate test data: 
```
python Prepare_Neg_test.py
```
It should be noted that the number of images in the Prepare_Neg_test.py file needs to be modified.

*B.Pretrained weights* 
The pretrained weights pth file can be obtained from 
[Google Drive](https://drive.google.com/file/d/1LffUUhT1eiSVeLAOlLCg_cBbOjSaaows/view?usp=sharing). 
[Baidu Drive(btsd)](https://pan.baidu.com/s/1NjkumS8LaHJtTbOZxfqtkQ)

**2. Train the MHA-UNet.** 
```
python train.py
```
- After trianing, you could obtain the outputs in './results/'

**3. Test the MHA-UNet.** 
First, in the test.py file, you should change the address of the checkpoint in 'resume_model' and fill in the location of the test data in 'data_path'.
```
python test.py
```
- After testing, you could obtain the outputs in './results/'

**4. Get the MHA-UNet explainable results map and EICA calculations.** 
First, in the test_Explainable.py file, you should change the address of the checkpoint in 'resume_model' and fill in the location of the test data in 'data_path'.
```
python test_Explainable.py
```
- After testing, you could obtain the outputs in './results/'. EICA is calculated for each case. EICA threshold defaults to 225. The final display 'Detected as true(number):' is the number of all detected as positive.

## Citation
If you find this repository helpful, please consider citing:
```
@INPROCEEDINGS{11356752,
  author={Wu, Renkai and Liu, Junwei and Liu, Yinghao and Liang, Pengchen and Ning, Guochen and Chang, Qing},
  booktitle={2025 IEEE International Conference on Bioinformatics and Biomedicine (BIBM)}, 
  title={Only Positive Cases: 5-Fold High-Order Spatial Attention Interaction Model for Skin Segmentation Derived Classification}, 
  year={2025},
  pages={4257-4262},
  doi={10.1109/BIBM66473.2025.11356752}}
}
```

## Acknowledgement 
This repo benefits from awesome works of [HorNet](https://github.com/raoyongming/HorNet), [MHorUNet](https://github.com/wurenkai/MHorUNet).

MHA-UNet

Explore Similar Repositories

Was this content helpful?

Recent Feedback

Download README