seann999 / ssd_tensorflow

SSD implementation in development in TensorFlow
182 stars 52 forks source link

How are PriorBoxLayer and MultiBoxLossLayer implemented? #7

Open alex-birch opened 7 years ago

alex-birch commented 7 years ago

Hi,

I'm quite new to tensorflow - could you please provide an explanation how you've implemented SSD without the 2 custom layers defined in Caffe - PriorBoxLayer and MultiBoxLossLayer?

From my (limited) understanding - MultiBoxLossLayer is needed for training and backpropagation connecting the ground truth boxes to intermediate layers and loss calculation. PriorBox is used in the forward pass only to generate boxes at the predefined sizes / aspect ratios.

Thanks