check pre-commit

pull/140/head
Ninenox 2 years ago
parent 2fb1e5dd8c
commit 8ce0d8bdba
  1. 1
      models/torch_utils.py

@ -73,6 +73,7 @@ def det_postprocess(data: Tuple[Tensor, Tensor, Tensor, Tensor]):
bboxes = bboxes[:nums] bboxes = bboxes[:nums]
scores = scores[:nums] scores = scores[:nums]
labels = labels[:nums] labels = labels[:nums]
# add nms # add nms
idx = nms(bboxes, scores, iou_thres) idx = nms(bboxes, scores, iou_thres)
bboxes, scores, labels = bboxes[idx], scores[idx], labels[idx] bboxes, scores, labels = bboxes[idx], scores[idx], labels[idx]

Loading…
Cancel
Save