search

facebookarchive / caffe2

Caffe2 is a lightweight, modular, and scalable deep learning framework.
https://caffe2.ai
Apache License 2.0
8.42k stars 1.95k forks source link

Semantic Segmentation Loss Function #2514

Open fatemeh-slh opened 6 years ago

fatemeh-slh commented 6 years ago

Hi,

I am trying to implement Deeplab model for semantic segmentation in Caffe2. I tried to use SigmoidCrossEntropyWithLogits as the loss function but I got this warning:

WARNING:root:You are creating an op that the ModelHelper does not recognize: SigmoidCrossEntropyWithLogits.

And it seems that the training process does not run correctly.

Does anybody try to implement a semantic segmentation (pre-trained on ImageNet) code in Caffe2?

Thanks.

beichen2012 commented 6 years ago

the softmax and loss function is: spatial_softmax_with_loss_op.