torrvision / crfasrnn

This repository contains the source code for the semantic image segmentation method described in the ICCV 2015 paper: Conditional Random Fields as Recurrent Neural Networks. http://crfasrnn.torr.vision/
Other
1.34k stars 462 forks source link

Any CUDA/GPU Implementation for CRFasRNN? #72

Closed lynetcha closed 7 years ago

lynetcha commented 8 years ago

Hi, is there a GPU/CUDA implementation for the multi_stage_meanfield layer? I can't find a multi_stage_meanfield.cu file or associated forward_gpu() methods in the caffe code.

chitrang6 commented 8 years ago

You have ti use the Caffe build provided in the Git repo. thank you