You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Amir 8f3b51b1bc first run 6 months ago
datasets delete python cache 6 months ago
models delete python cache 6 months ago
utils first run 6 months ago
.gitignore gitignore 6 months ago
README.md Update README.md 1 year ago
opt.py first run 6 months ago
results.txt first run 6 months ago
test.py Add files via upload 1 year ago
train.py first run 6 months ago

README.md

Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation

Requirements

  • torch
  • torchvision
  • scipy
  • PIL
  • numpy
  • tqdm

Data

$ data
train
├── Images
├── Masks
valid
├── Images
├── Masks
test
├── Images
├── Masks

1. Training

python train.py  --mode train  --train_data_dir /path-to-train_data  --valid_data_dir  /path-to-valid_data

2. Testing

```bash python test.py --mode test --load_ckpt checkpoint --test_data_dir /path-to-test_data