mxnet-ssd icon indicating copy to clipboard operation
mxnet-ssd copied to clipboard

Why number of labels has to be greater than 2 in MultiBoxTarget?

Open nttstar opened this issue 7 years ago • 4 comments

I saw one piece of code in MultiBoxTarget.cu below:

const int num_labels = labels.size(1);
CHECK_GT(num_labels, 2);

But why? I think one or two ground truth boxes are also valid for training.

nttstar avatar Jun 06 '18 06:06 nttstar

Just to make reuse of some temporary buffer without malloc new space.

zhreshold avatar Jun 06 '18 20:06 zhreshold

So we have to append '-1's if there are only 1 or 2 labels?

nttstar avatar Jun 07 '18 01:06 nttstar

correct

zhreshold avatar Jun 07 '18 02:06 zhreshold

I am also training on a different dataset and face this issue. Where do I append '-1'?

rkpandya avatar Sep 28 '18 21:09 rkpandya