# In[1]: import cv2 # In[2]: from models import ModelType, ModelFactory # In[3]: model = ModelFactory(rgbpath='trained_models/rgblstm.h5', trained=True).getModel(ModelType.RGB) # In[4]: model.summary() # In[5]: from keras.layers import Input from keras.models import Model # In[6]: rgbinput = Input((150, 100, 3)) x = model.layers[1].layer(rgbinput) for layer in model.layers[2:-3]: x = layer.layer(x) x