Pretrained TensorFlow Models about cascaded-fcn HOT 5 CLOSED

Do let me know if you get it.

from cascaded-fcn.

@alizeshana I forked that tensorflow-implementation branch and successfully started the training but then I got memory error because I only have one gtx 1060 6G card. If you have more powerful gpus maybe you can train it yourself

from cascaded-fcn.

@zakizhou I am facing the same issue on my gtx 740M 2GB. :p

Do let me know if you find the trained model.

from cascaded-fcn.

I think it's not very hard to convert the .caffeweights provided in this repo to a Keras .hdf5.

This is roughly how I would do it :

1. Define the same network in Keras (this may be the most time consuming). Make sure there are no implementation differences for any of the layers between Caffe and Keras.
2. A script Loads the .caffeweights and prototxt using caffe python library
3. Compile the keras architecture
4. Use keras's model.layers[i].set_weights(caffeweights), while figuring out any reshaping needed. Do this for every layer.
5. Save Keras's model as .hdf5 model.save_weights(...)

from cascaded-fcn.

If you manage to do this, I would appreciate sharing your work with others by submitting a PR with the Tensorflow/Keras weights.

from cascaded-fcn.