# 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