Предварительно обученные модели в Керасе, TorchVision - PullRequest
0 голосов
/ 06 марта 2019

У меня есть следующий код, который использует предварительно обученную ResNet50 модель в Керасе с imagenet DataSet:

from keras.applications.resnet50 import ResNet50
from keras.preprocessing import image
from keras.applications.resnet50 import preprocess_input, decode_predictions
import numpy as np

model = ResNet50(weights='imagenet')
print(model)

и все работает нормально. Мой вопрос: могу ли я найти предварительно обученную модель внутри Keras, Torchvision или TensorFlow, но для одного из следующих:

1) LeNet5 for MNIST DataSet
2) 32-Layer ResNet for the CIFAR-10 Dataset

Я знаю, что альтернатива состоит в том, чтобы обучить LeNet5, например, самостоятельно, но предпочтительнее будет предварительно обученная модель, и, насколько я искал, я не нашел их. спасибо.

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