Bienqq
/
car-detection-bayes
2
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

updates

This commit is contained in:
Glenn Jocher 2019-02-18 17:48:35 +01:00
parent fa0cbca69a
commit 6e2cf074a1
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
utils/utils.py
View File

@ -294,7 +294,7 @@ def build_targets(target, anchor_wh, nA, nC, nG):
def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4): def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4):
""" """
Removes detections with lower object confidence score than 'conf_thres' and performs Removes detections with lower object confidence score than 'conf_thres'
Non-Maximum Suppression to further filter detections. Non-Maximum Suppression to further filter detections.
Returns detections with shape: Returns detections with shape:
(x1, y1, x2, y2, object_conf, class_score, class_pred) (x1, y1, x2, y2, object_conf, class_score, class_pred)
@ -369,7 +369,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.4):
if prediction.is_cuda: if prediction.is_cuda:
unique_labels = unique_labels.cuda(prediction.device) unique_labels = unique_labels.cuda(prediction.device)
nms_style = 'OR' # 'AND' or 'OR' (classical) nms_style = 'OR' # 'AND', 'OR' (classical), 'MERGE' (experimental)
for c in unique_labels: for c in unique_labels:
# Get the detections with the particular class # Get the detections with the particular class
det_class = detections[detections[:, -1] == c] det_class = detections[detections[:, -1] == c]