Как найти слой переноса из переобученной начальной модели в Tensorflow - PullRequest
0 голосов
/ 13 марта 2020

Я закончил старое учебное пособие Tensorflow For Poets в Tensorflow версии 1. Я переобучил начальную модель, чтобы идентифицировать обрезанные изображения лиц.

Я могу видеть результаты как список вероятностей, но мой вопрос:

Как я могу получить доступ к обучающему слою (слою перед слоем Softmax)?

with tf.compat.v1.Session(graph=graph) as sess:
  results = sess.run(output_operation.outputs[0], {
    input_operation.outputs[0]: t
  })
results = np.squeeze(results)

top_k = results.argsort()[-5:][::-1]
labels = load_labels(label_file)
for i in top_k:
  print(labels[i], results[i])

Потому что, как только я получу на тренировочном уровне я хочу использовать уменьшение размерности (PCA и / или t-SNE), чтобы создать большой набор данных изображений с одним лицом, а затем использовать KNN (K-ближайших соседей), чтобы найти закрытое совпадение.

Вот полный код для возврата вероятностей .

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...