diff --git a/data/coco_64img.data b/data/coco_64img.data new file mode 100644 index 00000000..8fceee7f --- /dev/null +++ b/data/coco_64img.data @@ -0,0 +1,6 @@ +classes=80 +train=./data/coco_32img.txt +valid=./data/coco_32img.txt +names=data/coco.names +backup=backup/ +eval=coco diff --git a/data/coco_64img.txt b/data/coco_64img.txt new file mode 100644 index 00000000..306ff3b4 --- /dev/null +++ b/data/coco_64img.txt @@ -0,0 +1,64 @@ +../coco/images/train2014/COCO_train2014_000000000009.jpg +../coco/images/train2014/COCO_train2014_000000000025.jpg +../coco/images/train2014/COCO_train2014_000000000030.jpg +../coco/images/train2014/COCO_train2014_000000000034.jpg +../coco/images/train2014/COCO_train2014_000000000036.jpg +../coco/images/train2014/COCO_train2014_000000000049.jpg +../coco/images/train2014/COCO_train2014_000000000061.jpg +../coco/images/train2014/COCO_train2014_000000000064.jpg +../coco/images/train2014/COCO_train2014_000000000071.jpg +../coco/images/train2014/COCO_train2014_000000000072.jpg +../coco/images/train2014/COCO_train2014_000000000077.jpg +../coco/images/train2014/COCO_train2014_000000000078.jpg +../coco/images/train2014/COCO_train2014_000000000081.jpg +../coco/images/train2014/COCO_train2014_000000000086.jpg +../coco/images/train2014/COCO_train2014_000000000089.jpg +../coco/images/train2014/COCO_train2014_000000000092.jpg +../coco/images/train2014/COCO_train2014_000000000094.jpg +../coco/images/train2014/COCO_train2014_000000000109.jpg +../coco/images/train2014/COCO_train2014_000000000110.jpg +../coco/images/train2014/COCO_train2014_000000000113.jpg +../coco/images/train2014/COCO_train2014_000000000127.jpg +../coco/images/train2014/COCO_train2014_000000000138.jpg +../coco/images/train2014/COCO_train2014_000000000142.jpg +../coco/images/train2014/COCO_train2014_000000000144.jpg +../coco/images/train2014/COCO_train2014_000000000149.jpg +../coco/images/train2014/COCO_train2014_000000000151.jpg +../coco/images/train2014/COCO_train2014_000000000154.jpg +../coco/images/train2014/COCO_train2014_000000000165.jpg +../coco/images/train2014/COCO_train2014_000000000194.jpg +../coco/images/train2014/COCO_train2014_000000000201.jpg +../coco/images/train2014/COCO_train2014_000000000247.jpg +../coco/images/train2014/COCO_train2014_000000000260.jpg +../coco/images/train2014/COCO_train2014_000000000263.jpg +../coco/images/train2014/COCO_train2014_000000000307.jpg +../coco/images/train2014/COCO_train2014_000000000308.jpg +../coco/images/train2014/COCO_train2014_000000000309.jpg +../coco/images/train2014/COCO_train2014_000000000312.jpg +../coco/images/train2014/COCO_train2014_000000000315.jpg +../coco/images/train2014/COCO_train2014_000000000321.jpg +../coco/images/train2014/COCO_train2014_000000000322.jpg +../coco/images/train2014/COCO_train2014_000000000326.jpg +../coco/images/train2014/COCO_train2014_000000000332.jpg +../coco/images/train2014/COCO_train2014_000000000349.jpg +../coco/images/train2014/COCO_train2014_000000000368.jpg +../coco/images/train2014/COCO_train2014_000000000370.jpg +../coco/images/train2014/COCO_train2014_000000000382.jpg +../coco/images/train2014/COCO_train2014_000000000384.jpg +../coco/images/train2014/COCO_train2014_000000000389.jpg +../coco/images/train2014/COCO_train2014_000000000394.jpg +../coco/images/train2014/COCO_train2014_000000000404.jpg +../coco/images/train2014/COCO_train2014_000000000419.jpg +../coco/images/train2014/COCO_train2014_000000000431.jpg +../coco/images/train2014/COCO_train2014_000000000436.jpg +../coco/images/train2014/COCO_train2014_000000000438.jpg +../coco/images/train2014/COCO_train2014_000000000443.jpg +../coco/images/train2014/COCO_train2014_000000000446.jpg +../coco/images/train2014/COCO_train2014_000000000450.jpg +../coco/images/train2014/COCO_train2014_000000000471.jpg +../coco/images/train2014/COCO_train2014_000000000490.jpg +../coco/images/train2014/COCO_train2014_000000000491.jpg +../coco/images/train2014/COCO_train2014_000000000510.jpg +../coco/images/train2014/COCO_train2014_000000000514.jpg +../coco/images/train2014/COCO_train2014_000000000529.jpg +../coco/images/train2014/COCO_train2014_000000000531.jpg diff --git a/train.py b/train.py index ab6cdd64..531edb95 100644 --- a/train.py +++ b/train.py @@ -324,7 +324,7 @@ if __name__ == '__main__': # Mutate hyperparameters old_hyp = hyp.copy() init_seeds(seed=int(time.time())) - s = [.2, .2, .2, .2, .2, .3, .2, .2, .01, .3] + s = [.2, .2, .2, .2, .3, .2, .2, .02, .3] for i, k in enumerate(hyp.keys()): x = (np.random.randn(1) * s[i] + 1) ** 1.1 # plt.hist(x.ravel(), 100) hyp[k] = hyp[k] * float(x) # vary by about 30% 1sigma