https://github.com/datvodinh/neopolyp

Colonoscopy Polyp Segmentation and Neoplasm Characterization using U-Net
https://github.com/datvodinh/neopolyp

attention-mechanism resnet unet

Last synced: about 17 hours ago
JSON representation

Colonoscopy Polyp Segmentation and Neoplasm Characterization using U-Net

• Host: GitHub
• URL: https://github.com/datvodinh/neopolyp
• Owner: datvodinh
• Created: 2023-11-09T05:36:44.000Z (about 1 year ago)
• Default Branch: master
• Last Pushed: 2024-01-16T05:32:15.000Z (10 months ago)
• Last Synced: 2024-01-16T07:36:26.782Z (10 months ago)
• Topics: attention-mechanism, resnet, unet
• Language: Python
• Homepage:
• Size: 490 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

        
![Alt text](img/visualize.png)

## Setup

```bash

source ./script/setup.sh

```

## Training

```bash

source ./script/train.sh

```

## Testing

```bash

python -m neopolyp.infer --model [CHECKPOINT_PATH] --data_path [DATA_PATH] --save_path [SAVE_PATH]

```