From a77986fd7481a05a42e0880b973701244d0ceba9 Mon Sep 17 00:00:00 2001 From: Sayak Paul Date: Sat, 26 Oct 2019 09:21:54 +0530 Subject: [PATCH] Fixed ResNet18 import --- libs/models_keras.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/libs/models_keras.py b/libs/models_keras.py index 3545bcd..1556486 100644 --- a/libs/models_keras.py +++ b/libs/models_keras.py @@ -40,8 +40,7 @@ def build_unet(size=300, basef=64, maxf=512, encoder='resnet50', pretrained=True def make_encoder(input, name='resnet50', pretrained=True): if name == 'resnet18': - from classification_models.keras import Classifiers - ResNet18, _ = Classifiers.get('resnet18') + from keras.applications.resnet import ResNet18 model = ResNet18( weights='imagenet' if pretrained else None, input_tensor=input,