IBBM / Cascaded-FCN

Source code for the MICCAI 2016 Paper "Automatic Liver and Lesion Segmentation in CT Using Cascaded Fully Convolutional NeuralNetworks and 3D Conditional Random Fields"
Other
304 stars 127 forks source link

Issue with prediction method #32

Open nerdykamil opened 6 years ago

nerdykamil commented 6 years ago

OS: ubuntu 16.04 GPU: Nvidia GeForce GTX 105 The kernel dies while loading network when working from docker and when I set up my own environment I am able to run it through some layers only from net.forward() method but unable to get to the output, the kernel dies on forward pass. kindly suggest how to solve the issue as I am testing it only for one image right now as given in the notebook. Thank you

screenshot from 2018-07-05 18-33-41