Browse Source

Create README.md

main
rucv 3 years ago
parent
commit
4663424e49
No account linked to committer's email address
1 changed files with 28 additions and 0 deletions
  1. 28
    0
      README.md

+ 28
- 0
README.md View File

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

## Requirements

* torch
* torchvision
* scipy
* PIL
* numpy
* tqdm

### Data





### 1. Training

```bash
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```
```

Loading…
Cancel
Save