python - 如何使用我自己的数据在“Floydhub”上运行“Pix2Pix”代码的训练/测试命令？

Question

我正在尝试使用我自己的数据在 Floydhub 上运行Pix2Pix-tensorflow 。

我已经按照 Pix2Pix 文档中的说明组织了我的数据文件夹。数据文件夹包含 3 个文件夹：train、test、val。

我试图在 floydhub 命令行上运行以下 train 命令：

floyd run --gpu --data zaalee/datasets/tablesdata/2:/my_data "python pix2pix.py --mode train --output_dir /output --max_epochs 200 --input_dir /my_data --which_direction AtoB"

但是，在运行作业几秒钟后，作业失败，我收到以下错误：

2019-03-24 20:25:46,420 INFO - Traceback (most recent call last):

2019-03-24 20:25:46,420 INFO - File "pix2pix.py", line 803, in <module>

2019-03-24 20:25:46,420 INFO - main()

2019-03-24 20:25:46,420 INFO - File "pix2pix.py", line 625, in main

2019-03-24 20:25:46,421 INFO - examples = load_examples()

2019-03-24 20:25:46,421 INFO - File "pix2pix.py", line 244, in load_examples

2019-03-24 20:25:46,421 INFO - raise Exception("input_dir contains no image files")

2019-03-24 20:25:46,421 INFO - Exception: input_dir contains no image files

2019-03-24 20:25:50,680 INFO -

Answer 1

想通了我的错误。

正确的命令是——

floyd run --gpu --data zaalee/datasets/tablesdata/2:/my_data "python pix2pix.py --mode train --output_dir /output --max_epochs 200 --input_dir /my_data/train --which_direction AtoB"

需要在文件夹之后添加“火车”文件my_data夹，如下所示 --input_dir /my_data/train