Предварительно обученные контрольные точки (.chkpt) и GraphDef (.pb)

Я пытаюсь извлечь возможности нового набора данных с помощью предварительно обученной сети, подобной той, что classify_image_graph_def.pb, выпущенной Google в программе tenorflow (inception-2015-12-05.tgz). Я был успешен в этом, так как есть инструкция на Transfer_learning, которая использует classify_image_graph_def.pb (inception_v3.pb), чтобы извлечь переломы из нового набора данных.

Однако в новом выпуске предварительно обученных моделей тензор потока предоставляет файлы контрольных точек (например, resnet_v1_152.ckpt) вместо Graph_def (например, resnet_v1_152.pb). Мне было интересно, как я могу использовать эти файлы контрольных точек для извлечения функций, как в Transfer_learning. Кто-нибудь может дать мне несколько указаний?

1 ответ

Просто следуйте официальной модели сохранения / восстановления документации здесь.

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