Экспортный оценщик TF - PullRequest
Новая
       7

Экспортный оценщик TF

0 голосов
/ 22 февраля 2019

Я пытаюсь экспортировать оценку в tensorflow, но я застрял и не могу сделать это правильно.Сам оценщик создан из TensorFlowHub MobilenetV2 (https://tfhub.dev/google/imagenet/mobilenet_v2_035_224/feature_vector/2) + добавлен один плотный слой 

estimator = tf.estimator.Estimator(model_fn, params=params, config=config)
estimator.train(lambda: batched_train_input_fn(args.data_dir, params))
features = {'in': tf.placeholder(tf.float32, [1, 224, 224, 3], name="in")}
ex_input_fn = estimator.export.build_raw_serving_input_receiver_fn(features, 1)
exported_model_path = estimator.export_savedmodel(os.path.join(args.model_dir, 'exported'), ex_input_fn)

функция самой модели: 

def build_model(is_training, images, params):
    import tensorflow_hub as hub
    module = hub.Module("https://tfhub.dev/google/imagenet/mobilenet_v2_035_224/feature_vector/2")

    tf_model = module(images)
    with tf.variable_scope('fc_1'):
        tf_model = tf.layers.dense(tf_model, params.embedding_size)
    return model

Я получаю ошибку:

AttributeError: у объекта «Оценщик» нет атрибута «export»

Я думаю, что export_savedmodel является нормальной функцией в tf.

Где у меня проблема?

...