Попытка оптимизировать граф с помощью optimize_for_inference слишком много позиционных аргументов

Я делаю некоторое распознавание изображений с tensorflow, Я переучил CNN основанный на Inception V3 с моими данными. Мои результаты довольно хороши, я сейчас пытаюсь оптимизировать protobuf, чтобы использовать его с TFmobile.

Поэтому я использую optimize_for_inference, как это:

python optimize_for_inference.py --input = "C: \ Users \ project \ Documents \ peinture \ learn \ retrained_graphpeinture.pb" --output = "C: \ Users \ project \ Documents \ peinture \ learn \ optimized_graph.pb" - frozen_graph="True" --input_names="Заполнитель" --output_names="final_result"

В ответ я получил сообщение об ошибке:

 File "optimize_for_inference.py", line 92, in main
FLAGS.toco_compatible)
TypeError: optimize_for_inference() takes 4 positional arguments but 5 were given

Я понимаю, что в файле py слишком много аргументов. Я посмотрел на строку файла 92, и там действительно 5 аргументов:

output_graph_def = optimize_for_inference_lib.optimize_for_inference(
  input_graph_def,
  FLAGS.input_names.split(","),
  FLAGS.output_names.split(","),
  _parse_placeholder_types(FLAGS.placeholder_type_enum),
  FLAGS.toco_compatible)

Мой файл не хороший? Я не понимаю, почему это не работает. Я почти уверен, что использую хороший файл (клон формы Github). Я шаг за шагом следовал известному учебнику Tensorflow для поэтов 2.

С наилучшими пожеланиями

0 ответов

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