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

updates

This commit is contained in:
Glenn Jocher 2019-09-27 23:36:42 +02:00
parent b421afa508
commit 286e851fe7
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -451,7 +451,7 @@ def non_max_suppression(prediction, conf_thres=0.5, nms_thres=0.5):
min_wh = 2 # (pixels) minimum box width and height min_wh = 2 # (pixels) minimum box width and height
output = [None] * len(prediction) output = [None] * len(prediction)
for image_i, pred in enumerate(prediction): for image_i, pred in enumerate(prediction.float()):
# Experiment: Prior class size rejection # Experiment: Prior class size rejection
# x, y, w, h = pred[:, 0], pred[:, 1], pred[:, 2], pred[:, 3] # x, y, w, h = pred[:, 0], pred[:, 1], pred[:, 2], pred[:, 3]
# a = w * h # area # a = w * h # area