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

updates

This commit is contained in:
Glenn Jocher 2018-08-26 10:59:39 +02:00
parent 5463ab8aa0
commit be17c9aaf6
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/utils.py
View File

@ -30,7 +30,6 @@ def modelinfo(model):
print('\n%g layers, %g parameters, %g gradients' % (i + 1, nparams, ngradients)) print('\n%g layers, %g parameters, %g gradients' % (i + 1, nparams, ngradients))
def xview_class_weights(indices): # weights of each class in the training set, normalized to mu = 1 def xview_class_weights(indices): # weights of each class in the training set, normalized to mu = 1
weights = 1 / torch.FloatTensor( weights = 1 / torch.FloatTensor(
[74, 364, 713, 71, 2925, 209767, 6925, 1101, 3612, 12134, 5871, 3640, 860, 4062, 895, 149, 174, 17, 1624, 1846, [74, 364, 713, 71, 2925, 209767, 6925, 1101, 3612, 12134, 5871, 3640, 860, 4062, 895, 149, 174, 17, 1624, 1846,
@ -40,7 +39,6 @@ def xview_class_weights(indices): # weights of each class in the training set,
return weights[indices] return weights[indices]
def plot_one_box(x, im, color=None, label=None, line_thickness=None): def plot_one_box(x, im, color=None, label=None, line_thickness=None):
tl = line_thickness or round(0.003 * max(im.shape[0:2])) # line thickness tl = line_thickness or round(0.003 * max(im.shape[0:2])) # line thickness
color = color or [random.randint(0, 255) for _ in range(3)] color = color or [random.randint(0, 255) for _ in range(3)]