diff --git a/utils/utils.py b/utils/utils.py index 67eaed19..79737fb3 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -30,7 +30,6 @@ def modelinfo(model): 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 weights = 1 / torch.FloatTensor( [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] - 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 color = color or [random.randint(0, 255) for _ in range(3)]