carlthome / tensorflow-convlstm-cell

A ConvLSTM cell with layer normalization and peepholes for TensorFlow's RNN API.
MIT License
399 stars 150 forks source link

How to change this class to Keras Layer? #35

Open smoein opened 4 years ago

smoein commented 4 years ago

I want to make this deep learning network with Keras. This network is proposed for compressing video image One layer of this model is ConvLSTM. ConvLSTM is good for compressing sequences of images. I know Keras has the ConvLSTM2D layer but I want to use the ConvLSTM cell. Now I don't know how to change this class to a custom Keras Layer. Can anyone help me? @xdurch0

NoviceMAn-prog commented 3 years ago

Did you find any solution to this?