Попытка оптимизировать граф с помощью 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.
С наилучшими пожеланиями