未验证 提交 5d172a5f 编写于 作者: W whs 提交者: GitHub

Fix readme of cycle gan. (#1300)

* Fix readme of cycle gan.

* Fix doc style.
上级 96bb9a30
...@@ -21,21 +21,23 @@ TODO ...@@ -21,21 +21,23 @@ TODO
horse2zebra训练集包含1069张野马图片,1336张斑马图片。测试集包含121张野马图片和141张斑马图片。 horse2zebra训练集包含1069张野马图片,1336张斑马图片。测试集包含121张野马图片和141张斑马图片。
数据下载处理完毕后,并组织为以下路径: 数据下载处理完毕后,并组织为以下路径结构
``` ```
horse2zebra/ data
|-- testA |-- horse2zebra
|-- testA.txt | |-- testA
|-- testB | |-- testA.txt
|-- testB.txt | |-- testB
|-- trainA | |-- testB.txt
|-- trainA.txt | |-- trainA
|-- trainB | |-- trainA.txt
`-- trainB.txt | |-- trainB
| `-- trainB.txt
``` ```
以上数据文件中,‘testA’为存放野马测试图片的文件夹,‘testB’为存放斑马测试图片的文件夹,'testA.txt'和'testB.txt'分别为野马和斑马测试图片路径列表文件,格式如下: 以上数据文件中,`data`文件夹需要放在训练脚本`train.py`同级目录下。`testA`为存放野马测试图片的文件夹,`testB`为存放斑马测试图片的文件夹,`testA.txt``testB.txt`分别为野马和斑马测试图片路径列表文件,格式如下:
``` ```
testA/n02381460_9243.jpg testA/n02381460_9243.jpg
...@@ -53,7 +55,7 @@ testA/n02381460_9245.jpg ...@@ -53,7 +55,7 @@ testA/n02381460_9245.jpg
在GPU单卡上训练: 在GPU单卡上训练:
``` ```
env CUDA_VISIABLE_DEVICES=0 python train.py env CUDA_VISIBLE_DEVICES=0 python train.py
``` ```
执行`python train.py --help`可查看更多使用方式和参数详细说明。 执行`python train.py --help`可查看更多使用方式和参数详细说明。
...@@ -72,7 +74,7 @@ env CUDA_VISIABLE_DEVICES=0 python train.py ...@@ -72,7 +74,7 @@ env CUDA_VISIABLE_DEVICES=0 python train.py
``` ```
env CUDA_VISIBLE_DEVICE=0 python infer.py \ env CUDA_VISIBLE_DEVICE=0 python infer.py \
--model_path="models/1" --input="./data/inputA/*" \ --init_model="models/1" --input="./data/inputA/*" \
--output="./output" --output="./output"
``` ```
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册