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

updates

This commit is contained in:
Glenn Jocher 2019-12-30 13:15:10 -08:00
parent e4a797fc1e
commit 121526aa98
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

@ -746,7 +746,7 @@ def kmeans_targets(path='data/coco64.txt', n=9, img_size=416): # from utils.uti
l[:, [2, 4]] *= s[1] l[:, [2, 4]] *= s[1]
l[:, 1:] *= img_size / max(s) l[:, 1:] *= img_size / max(s)
l = l.repeat(10, axis=0) # augment 10x l = l.repeat(10, axis=0) # augment 10x
l *= np.random.uniform(0.5, 1.5, size=(l.shape[0], 1)) # multi-scale box l *= np.random.uniform(288, 640, size=(l.shape[0], 1)) / img_size # multi-scale box
wh = np.concatenate(dataset.labels, 0)[:, 3:5] # wh from cxywh wh = np.concatenate(dataset.labels, 0)[:, 3:5] # wh from cxywh
# Kmeans calculation # Kmeans calculation