search

sadeepj / crfasrnn_keras

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

GPU support for 3D spatial data/ND channel data images #42

Closed heyitsguay closed 6 years ago

heyitsguay commented 6 years ago

Same as the pull request on master, but with GPU support. Had to do a bit more work to get it working, but as of this morning I'm getting the same results from the CPU and GPU branches of my fork.

heyitsguay commented 6 years ago

Just noticed the RGB demo needs to be reworked for the GPU version. Will fix and create a new pull request.