simple-ssd-for-beginners

simple-ssd-for-beginners copied to clipboard

Hello amazing work I had one question how did you calculate the scales in config.py `(30, 60, 111, 162, 213, 264, 315)` what `Smin` and `Smax` did you use?

PushpakBhoge512

Hi, I am using this implementation for a custom dataset. While running the training loop, I get an error from the line: `loc_loss, cls_loss = criterion(p_loc, p_label, gt_boxes, gt_labels)` The...

madiltalay

nms这块

2

``` # 对每一类进行检测 for i in range(scores.shape[1]): # 得到这一类中可能有物体的anchor mask = scores[:, i] >= gt_threshold label_scores = scores[mask, i] label_boxes = locations[mask] # 没有找到直接下一个类别 if len(label_scores) == 0: continue #...

HiIcy