Where can I find a information about stage weight in the original paper?

zhaoweicai / cascade-rcnn

Caffe implementation of multiple popular object detection frameworks

1.04k stars 293 forks source link

Closed kye9216789 closed 5 years ago

kye9216789 commented 5 years ago

Hi. I am finding about stage loss weights but I can not. How did you set the stage weights, (1.0, 0.5, 0.25)?

zhaoweicai commented 5 years ago

They are in the train.prototxt file. The parameter loss_weight assigns weights to the stage loss layers.