Piotr Chudzik

I second that!

@bleakie @rydenisbak @biubug6 Did anyone manage to make it work with more classes? I modified the class branch to take into account more classes: self.conv1x1 = nn.Conv2d(inchannels,self.num_anchors*num_classes,kernel_size=(1,1),stride=1,padding=0) However my loss...

To answer my question - yes nr of prior boxes must match wiith class head outputs. The problem was in reshaping the ClassHead output according to nr of classes: return...

@biubug6 If you could do it, that would be amazing!