martinkersner / train-CRF-RNN

Train CRF-RNN for Semantic Image Segmentation
Other
199 stars 92 forks source link

Segmentation fault #45

Open AlexTS1980 opened 7 years ago

AlexTS1980 commented 7 years ago

Both FCN and CRF-RNN report the segmentation fault error in the conv1_1 layer whenever I try finetuning with my own data. After some attempts I realized this is something to do with labels; when I comments out this layer, the network loads OK until the loss. I coundn't find any fix so far. Do you know anything about this? Data are images Bx3x250x250, labels are also images Bx1x250x250