search

sadeepj / crfasrnn_keras

CRF-RNN Keras/Tensorflow version
http://crfasrnn.torr.vision
MIT License
603 stars 169 forks source link

Keras from Tensorflow Version #78

Open mavaylon1 opened 3 years ago

mavaylon1 commented 3 years ago

Will there be an update where it uses keras from tensorflow and not keras the api?

YeonghyeonKim commented 2 years ago

In src/crfrnn_layer.py,

from keras.engine.topology import Layer

=>

from tensorflow.keras.layers import Layer

In src/crfrnn_model.py,

from keras.models import Model
from keras.layers import Conv2D, MaxPooling2D, Input, ZeroPadding2D, \

=> 

from tensorflow.keras.models import Model
from tensorflow.keras.layers import Conv2D, MaxPooling2D, Input, ZeroPadding2D, \

Worked on me.