Проблема із запуском object_detection_tutorial TypeError: load () відсутні 2 необхідні позиційні аргументи

Я досить новачок у tensorflow і намагаюся запустити object_detection_tutorial. Я отримую TypeErrror і не знаю, як це виправити.

Це функція load_model, яка пропускає 2 аргументи:

Теги: Набір тегів рядків для ідентифікації необхідного MetaGraphDef. Вони повинні відповідати тегам, які використовуються під час збереження змінних за допомогою API SavedModel save ().

export_dir: Каталог, в якому розміщений буфер протоколу SavedModel та змінні, які потрібно завантажити.

def load_model(model_name):
  base_url = 'http://download.tensorflow.org/models/object_detection/'
  model_file = model_name + '.tar.gz'
  model_dir = tf.keras.utils.get_file(
    fname=model_name, 
    origin=base_url + model_file,
    untar=True)

  model_dir = pathlib.Path(model_dir)/"saved_model"

  model = tf.saved_model.load(str(model_dir))
  model = model.signatures['serving_default']

  return model

WARNING:tensorflow:From <ipython-input-9-f8a3c92a04a4>:11: load (from tensorflow.python.saved_model.loader_impl) is deprecated and will be removed in a future version.
Instructions for updating:
This function will only be available through the v1 compatibility library as tf.compat.v1.saved_model.loader.load or tf.compat.v1.saved_model.load. There will be a new function for importing SavedModels in Tensorflow 2.0.

---------------------------------------------------------------------------
TypeError                                 Traceback (most recent call last)
<ipython-input-12-e10c73a22cc9> in <module>
      1 model_name = 'ssd_mobilenet_v1_coco_2017_11_17'
----> 2 detection_model = load_model(model_name)

<ipython-input-9-f8a3c92a04a4> in load_model(model_name)
      9   model_dir = pathlib.Path(model_dir)/"saved_model"
     10 
---> 11   model = tf.saved_model.load(str(model_dir))
     12   model = model.signatures['serving_default']
     13 

~/.local/lib/python3.6/site-packages/tensorflow/python/util/deprecation.py in new_func(*args, **kwargs)
    322               'in a future version' if date is None else ('after %s' % date),
    323               instructions)
--> 324       return func(*args, **kwargs)
    325     return tf_decorator.make_decorator(
    326         func, new_func, 'deprecated',

TypeError: load() missing 2 required positional arguments: 'tags' and 'export_dir'

Чи можете ви допомогти мені виправити це та запустити мій перший детектор об'єктів: D?

tensorflow

— Домінік
джерело

Відповіді:

У мене була така ж проблема, і я намагаюся вирішити це вже 1 тиждень. Я думаю, рішення повинно бути таким;

model = tf.compat.v2.saved_model.load(str(model_dir), None)

Більш детально було б (з офіційного веб-сайту );

Завантажте збережену модель з export_dir.

tf.saved_model.load(
    export_dir,
    tags=None
)

Псевдоніми:

tf.compat.v1.saved_model.load_v2

tf.compat.v2.saved_model.load

— Онур Баскін
джерело

Я використав ваше рішення і отримав ще одну помилку. Я оновив усе, що міг, і це працює! У мене також виникла помилка з тим, що pathlib не будувався.

— Домінік

@Dominik Ви можете бути більш конкретними? можливо, я можу допомогти, тому що ця пригода на тензорфлоу я привела мене до вирішення багатьох питань: D

— Онур Баскін

@OnurBaskin Пізніше з'явилася помилка: Аргумент TypeError: int () повинен бути рядок,

— байтовий

@Dominik Я припускаю, що це ваша версія Tensorflow. Це має бути версія 2.0 (стабільна). Ось посилання на запитання, яке я задав, можливо у вас є точна помилка. Також знайдіть будь-який старий імпорт, для якого потрібен "compat.v1". пізніше у вас повинно бути багато більше помилок, але саме так ви мігруєте старий код.

— Онур Баскін

@OnurBaskin Я дуже розгублений. Я вважав, що API виявлення об'єктів сумісний лише з версіями TensorFlow 1.

— Biiiiiird

Я здогадався, що це проблема з гілкою, і використання гілки tf_2_1_reference зробило для мене хитрість:

igian@iGians-MBP models % git checkout tf_2_1_reference
M   research/object_detection/object_detection_tutorial.ipynb
Branch 'tf_2_1_reference' set up to track remote branch 'tf_2_1_reference' from 'origin'.
Switched to a new branch 'tf_2_1_reference'
igians@iGians-MBP models % jupyter notebook

Тоді виконуйте кожну клітинку юпітера підручника, як добрий новачок!

Це галузь, яку я використав: https://github.com/tensorflow/models/tree/tf_2_1_reference

— iGian
джерело