summaryrefslogtreecommitdiff
path: root/README.md
diff options
context:
space:
mode:
authorjunyanz <junyanz@berkeley.edu>2017-04-20 04:42:33 -0700
committerjunyanz <junyanz@berkeley.edu>2017-04-20 04:42:33 -0700
commitbce1bbd63c469b5fdec88abad614060c66cd7132 (patch)
tree2fabba4b6afadefabbcaeed8b0a2bc61affe26cd /README.md
parent2d4b9bf9149126d20237dc357a8c23294273e91b (diff)
update README
Diffstat (limited to 'README.md')
-rw-r--r--README.md10
1 files changed, 7 insertions, 3 deletions
diff --git a/README.md b/README.md
index 42ddcd3..d44ec79 100644
--- a/README.md
+++ b/README.md
@@ -60,13 +60,15 @@ cd pytorch-CycleGAN-and-pix2pix
```bash
bash ./datasets/download_cyclegan_dataset.sh maps
```
-- Train a model (`bash ./scripts/train_cyclegan.sh`):
+- Train a model:
```bash
+# ./scripts/train_cyclegan.sh
python train.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan
```
- To view training results and loss plots, run `python -m visdom.server` and click the URL http://localhost:8097. To see more intermediate results, check out `./checkpoints/maps_cyclegan/web/index.html`
-- Test the model (`bash ./scripts/test_cyclegan.sh`):
+- Test the model:
```bash
+#!./scripts/test_cyclegan.sh
python test.py --dataroot ./datasets/maps --name maps_cyclegan --model cycle_gan --phase test
```
The test results will be saved to a html file here: `./results/maps_cyclegan/latest_test/index.html`.
@@ -76,13 +78,15 @@ The test results will be saved to a html file here: `./results/maps_cyclegan/lat
```bash
bash ./datasets/download_pix2pix_dataset.sh facades
```
-- Train a model (`bash ./scripts/train_pix2pix.sh`):
+- Train a model:
```bash
+#!./scripts/train_pix2pix.sh
python train.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --lambda_A 100 --align_data --no_lsgan
```
- To view training results and loss plots, run `python -m visdom.server` and click the URL http://localhost:8097. To see more intermediate results, check out `./checkpoints/facades_pix2pix/web/index.html`
- Test the model (`bash ./scripts/test_pix2pix.sh`):
```bash
+#!./scripts/test_pix2pix.sh
python test.py --dataroot ./datasets/facades --name facades_pix2pix --model pix2pix --which_model_netG unet_256 --which_direction BtoA --align_data
```
The test results will be saved to a html file here: `./results/facades_pix2pix/latest_val/index.html`.