Browse Source

Update README.md

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

+ 3
- 3
README.md View File

@@ -1,6 +1,6 @@
# Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation
## Enhanced U-Net: A Feature Enhancement Network for Polyp Segmentation

## Requirements
### Requirements

* torch
* torchvision
@@ -29,7 +29,7 @@ test
python train.py --mode train --train_data_dir /path-to-train_data --valid_data_dir /path-to-valid_data
```

#### 2. Testing
### 2. Testing

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

Loading…
Cancel
Save