Skip to content
This repository was archived by the owner on Nov 9, 2022. It is now read-only.

ResNet18 import fixation #12

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

ResNet18 import fixation #12

wants to merge 1 commit into from

Conversation

sayakpaul
Copy link

Referring to the models_keras.py script, ResNet18 is being imported as from classification_models.keras import Classifiers but I was unable to find the classification_models module in the first place.

I swapped it with:

from keras.applications.resnet import ResNet18
model = ResNet18(
            weights='imagenet' if pretrained else None,
            input_tensor=input,
            include_top=False)

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Development

Successfully merging this pull request may close these issues.

1 participant