diff --git a/detect.py b/detect.py index 475a2321..b112bba7 100644 --- a/detect.py +++ b/detect.py @@ -1,5 +1,4 @@ import argparse -import time from sys import platform from models import * # set ONNX_EXPORT in models.py diff --git a/train.py b/train.py index 37e4ab13..24165fd9 100644 --- a/train.py +++ b/train.py @@ -430,7 +430,7 @@ if __name__ == '__main__': # Mutate init_seeds(seed=int(time.time())) - s = [.15, .15, .15, .15, .15, .15, .15, .00, .02, .20, .15, .0, .0, .0, .0, .0, .0] # sigmas + s = [.15, .15, .15, .15, .15, .15, .15, .00, .02, .20, .20, .20, .20, .20, .20, .20, .20] # sigmas for i, k in enumerate(hyp.keys()): x = (np.random.randn(1) * s[i] + 1) ** 2.0 # plt.hist(x.ravel(), 300) hyp[k] *= float(x) # vary by sigmas diff --git a/utils/utils.py b/utils/utils.py index 0202b8f5..4987fedb 100755 --- a/utils/utils.py +++ b/utils/utils.py @@ -88,6 +88,9 @@ def coco_class_weights(): # frequency of each class in coco train2014 1877, 17630, 4337, 4624, 1075, 3468, 135, 1380] weights = 1 / torch.Tensor(n) weights /= weights.sum() + # with open('data/coco.names', 'r') as f: + # for k, v in zip(f.read().splitlines(), n): + # print('%20s: %g' % (k, v)) return weights