Question

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

我已经按照Pix2Pix文档中的说明组织了数据文件夹。数据文件夹包含3个文件夹：训练，测试，验证。

我试图在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文件夹之后添加“ train”文件夹，就像--input_dir /my_data/train

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

1 个答案: