Модель Tensorflow GPU не может тренироваться на пользовательских изображениях - PullRequest
0 голосов
/ 28 июня 2018

Я недавно играл с обнаружением объектов тензорного потока с процессором GPU, и я столкнулся с ошибкой при попытке натренировать мою модель с помощью пользовательских изображений. Трассировка ошибки следующая:

 WARNING:tensorflow:From C:\tensorflow1\models\research\object_detection\trainer.py:260: create_global_step (from tensorflow.contrib.framework.python.ops.variables) is deprecated and will be removed in a future version.
Instructions for updating:
Please switch to tf.train.create_global_step
WARNING:tensorflow:num_readers has been reduced to 1 to match input file shards.
Traceback (most recent call last):
  File "train.py", line 184, in <module>
    tf.app.run()
  File "C:\Users\Dan\AppData\Local\conda\conda\envs\tensorflow1\lib\site-packages\tensorflow\python\platform\app.py", line 126, in run
    _sys.exit(main(argv))
  File "train.py", line 180, in main
    graph_hook_fn=graph_rewriter_fn)
  File "C:\tensorflow1\models\research\object_detection\trainer.py", line 274, in train
    train_config.prefetch_queue_capacity, data_augmentation_options)
  File "C:\tensorflow1\models\research\object_detection\trainer.py", line 80, in create_input_queue
    include_keypoints=include_keypoints))
  File "C:\tensorflow1\models\research\object_detection\core\preprocessor.py", line 3147, in preprocess
    (func.__name__))
ValueError: The function random_horizontal_flip does not exist in func_arg_map

Я использую интерпретатор Anaconda с Python 3.6, чтобы воспроизвести эту ошибку, я выполнил все шаги в ссылке https://github.com/EdjeElectronics/TensorFlow-Object-Detection-API-Tutorial-Train-Multiple-Objects-Windows-10.

Команда, которая выдала мне эту ошибку, была:

python train.py --logtostderr --train_dir=training/ --pipeline_config_path=training/faster_rcnn_inception_v2_pets.config

Важно отметить, что я не сталкивался с какими-либо проблемами до обучения модели. Я был бы рад, если бы кто-то мог объяснить мне эту ошибку и даже помочь мне исправить ее, спасибо заранее: -)

1 Ответ

0 голосов
/ 11 июля 2018

Вы можете забыть добавить следующую строку к ~/.bashrc: export PYTHONPATH=$PYTHONPATH: PWD : PWD /slim

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...