crfasrnn icon indicating copy to clipboard operation
crfasrnn copied to clipboard

training my own data

Open luciaL opened this issue 8 years ago • 5 comments

I use the crfrnn net to make a segmentation of an organ,so the number of output is 2.Then I modify the example/segmentationcrfrnn/solve.protoxt.Then I run the .sh file.But it check failed at softmax_loss_layer and said number of labels don't match number of predictions.Please help to figure out what's the problem??

Could you tell me how to make 'label' lmdb file?The name of label's size should be 121500500,how to make it?The number of my output is 2,so i need to make it be 12500500,but i don't know how to make it.Thank you very much

luciaL avatar Apr 28 '17 15:04 luciaL

You need to change the name of the deconvolution layer and you also need to change the number of the output.

Make sure you initialize the weights of all layers correctly.

bittnt avatar Apr 28 '17 18:04 bittnt

Could you tell me how to make 'label' lmdb file?The name of label's size should be 121500500,how to make it?The number of my output is 2,so i need to make it be 12500500,but i don't know how to make it.Thank you very much

luciaL avatar Apr 29 '17 16:04 luciaL

https://lmdb.readthedocs.io/en/release/ will probably help you.

bittnt avatar May 10 '17 23:05 bittnt

@bittnt i'm training with my own data, how to initialize the convolution weights?

lixiang-ucas avatar Oct 05 '17 09:10 lixiang-ucas

@lixiang-ucas I am also trying to use this network for organ segmentation, could you manage to train the network on your data? so far all my predictions are in black and I don't know whats the proper initialization of the convolutions/deconv?

Sam813 avatar Jun 04 '18 00:06 Sam813