search

GantMan / nsfw_model

Keras model of NSFW detector
Other
1.8k stars 279 forks source link

How to load model with CNN layers ? #90

Closed ducpvftech closed 3 years ago

ducpvftech commented 3 years ago

Thank you for this great repo.

The function load_model using custom_objects and return only 3 layers:

<tensorflow_hub.keras_layer.KerasLayer>
<tensorflow.python.keras.layers.core.Dense>
<tensorflow.python.keras.layers.core.Activation>

How I can load a model that contained CNN layer(s)?

Thank you so much!