Как выполнить команду train / test для кода «Pix2Pix» на «Floydhub», используя мои собственные данные?

Question

Я пытаюсь запустить Pix2Pix-tenorflow на Floydhub, используя мои собственные данные.

Я организовал свою папку данных, как указано в документации по Pix2Pix.Папка данных содержит 3 папки: train, test, val.

Я попытался запустить следующую команду train в командной строке floydhub:

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"

Папка "train" должна быть добавлена ​​после папки my_data примерно так --input_dir /my_data/train