Add an example notebook

Author: triple-Mu

models/common.py

@@ -985,7 +985,7 @@ def forward(self, x):
         labels = batched_labels[topk_batch_inds, topk_inds, ...]
         boxes, scores = batched_dets.split((4, 1), -1)
         scores = scores.squeeze(-1)
-        num_dets = (scores > 0).sum(1, keepdim=True)
+        num_dets = (labels > -1).sum(1, keepdim=True)
         return num_dets, boxes, scores, labels