Загрузка обученной сети ResNet 50 в Tensorflow в Windows

Question

Я недавно начал работать над Deep Learning.Мне нужно запустить предварительно обученную сеть ResNet50, загруженную с помощью Tensorflow на ЦП Windows.Я не нахожу структурированный способ сделать это где-нибудь в Интернете.Может кто-нибудь, пожалуйста, помогите мне с этим.

Answer 1

Вы должны следовать официальному учебному пособию «Как сделать Transfer Learning» с сайта tenorflow.https://www.tensorflow.org/hub/tutorials/image_retraining

То, что вы должны изменить, находится на retrain.py строке 1302, она будет загружать предварительно обученную модель из tenorflow-hub:

default=(
          'https://tfhub.dev/google/imagenet/inception_v3/feature_vector/1'),

в это:

default=(
          'https://tfhub.dev/google/imagenet/resnet_v1_50/feature_vector/1'),

Или вы можете проверить список других предварительно обученных моделей на этом сайте: https://tfhub.dev/s?module-type=image-feature-vector