rucv
3 years ago
No account linked to committer's email address
1 changed files with
28 additions and
0 deletions
-
README.md
|
|
|
|
|
|
|
|
|
|
|
# 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``` |
|
|
|
|
|
``` |