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

add stride order reversal for c53*.cfg

This commit is contained in:
Glenn Jocher 2020-05-17 22:11:02 -07:00
parent da40084b37
commit bc9da228e0
1 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
utils/utils.py
View File

@ -4,6 +4,7 @@ import os
import random import random
import shutil import shutil
import subprocess import subprocess
import time
from pathlib import Path from pathlib import Path
from sys import platform from sys import platform
@ -472,12 +473,14 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T
nx6 (x1, y1, x2, y2, conf, cls) nx6 (x1, y1, x2, y2, conf, cls)
""" """
# Box constraints # Settings
merge = True # merge for best mAP
min_wh, max_wh = 2, 4096 # (pixels) minimum and maximum box width and height min_wh, max_wh = 2, 4096 # (pixels) minimum and maximum box width and height
time_limit = 10.0 # seconds to quit after
t = time.time()
nc = prediction[0].shape[1] - 5 # number of classes nc = prediction[0].shape[1] - 5 # number of classes
multi_label &= nc > 1 # multiple labels per box multi_label &= nc > 1 # multiple labels per box
merge = True # merge for best mAP
output = [None] * prediction.shape[0] output = [None] * prediction.shape[0]
for xi, x in enumerate(prediction): # image index, image inference for xi, x in enumerate(prediction): # image index, image inference
# Apply constraints # Apply constraints
@ -533,6 +536,9 @@ def non_max_suppression(prediction, conf_thres=0.1, iou_thres=0.6, multi_label=T
pass pass
output[xi] = x[i] output[xi] = x[i]
if (time.time() - t) > time_limit:
break # time limit exceeded
return output return output