Ошибка: тензорный поток: не удалось понять имя архитектуры '.50_% image_size%'

Привет всем, кого я пытался создать классификатор изображений на основе TensorFlow для поэтов, я использую colab для запуска тензорного потока, но когда я запускаю приведенный ниже код

    IMAGE_SIZE=224
    ARCHITECTURE="mobilenet_0.50_${IMAGE_SIZE}"
    !python -m scripts.retrain \
      --bottleneck_dir=tf_files/bottlenecks \
      --how_many_training_steps=500 \
      --model_dir=tf_files/models/ \
      --summaries_dir=tf_files/training_summaries/"${ARCHITECTURE}" \
      --output_graph=tf_files/retrained_graph.pb \
      --output_labels=tf_files/retrained_labels.txt \
      --architecture="${ARCHITECTURE}" \
      --image_dir=tf_files/flower_photos

Я получаю следующую ошибку

ERROR:tensorflow:Couldn't understand architecture name '.50_%image_size%'
Traceback (most recent call last):
  File "/usr/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/usr/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/content/tensorflow-for-poets-2/scripts/retrain.py", line 1326, in <module>
    tf.app.run(main=main, argv=[sys.argv[0]] + unparsed)
  File "/usr/local/lib/python3.6/dist-packages/tensorflow/python/platform/app.py", line 125, in run
    _sys.exit(main(argv))
  File "/content/tensorflow-for-poets-2/scripts/retrain.py", line 976, in main
    model_info = create_model_info(FLAGS.architecture)
  File "/content/tensorflow-for-poets-2/scripts/retrain.py", line 923, in create_model_info
    raise ValueError('Unknown architecture', architecture)
ValueError: ('Unknown architecture', '.50_%image_size%')
0
0
192
1

Ответы 1

В ячейке кода есть две ошибки. Сначала вам нужно установить для переменной ARCΗΙTECTURE значение f"mobilenet_0.50_${IMAGE_SIZE}. Во-вторых, вы не должны использовать $ в команде оболочки. Вот код с исправлениями.

IMAGE_SIZE=224
ARCHITECTURE=f"mobilenet_0.50_${IMAGE_SIZE}"
!python -m scripts.retrain \
  --bottleneck_dir=tf_files/bottlenecks \
  --how_many_training_steps=500 \
  --model_dir=tf_files/models/ \
  --summaries_dir=tf_files/training_summaries/"{ARCHITECTURE}" \
  --output_graph=tf_files/retrained_graph.pb \
  --output_labels=tf_files/retrained_labels.txt \
  --architecture="{ARCHITECTURE}" \
  --image_dir=tf_files/flower_photos

Другие вопросы по теме