From 7e9e289b164e9bbfb77f644b1cbfac48bc8c8408 Mon Sep 17 00:00:00 2001 From: Edmond Yoo <hj3yoo@uwaterloo.ca> Date: Tue, 11 Sep 2018 00:51:08 +0000 Subject: [PATCH] training w/ full yolo cfg --- cfg/yolov3-custom.cfg | 12 ++++++------ 1 files changed, 6 insertions(+), 6 deletions(-) diff --git a/cfg/yolov3-custom.cfg b/cfg/yolov3-custom.cfg index a1681e5..e7d6a6e 100644 --- a/cfg/yolov3-custom.cfg +++ b/cfg/yolov3-custom.cfg @@ -1,7 +1,7 @@ [net] # Testing -batch=8 -subdivisions=4 +batch=64 +subdivisions=32 # Training # batch=64 # subdivisions=16 @@ -17,7 +17,7 @@ learning_rate=0.001 burn_in=1000 -max_batches = 500200 +max_batches = 30000 policy=steps steps=400000,450000 scales=.1,.1 @@ -606,7 +606,7 @@ [yolo] mask = 6,7,8 -anchors = 10,13, 16,30, 33,23, 30,61, 62,45, 59,119, 116,90, 156,198, 373,326 +anchors = 129.1513,150.1077, 104.9181,195.1675, 145.8067,186.4670, 127.9706,231.2296, 165.9494,221.8162, 162.4676,253.0225 classes=1 num=9 jitter=.3 @@ -692,7 +692,7 @@ [yolo] mask = 3,4,5 -anchors = 10,13, 16,30, 33,23, 30,61, 62,45, 59,119, 116,90, 156,198, 373,326 +anchors = 129.1513,150.1077, 104.9181,195.1675, 145.8067,186.4670, 127.9706,231.2296, 165.9494,221.8162, 162.4676,253.0225 classes=1 num=9 jitter=.3 @@ -779,7 +779,7 @@ [yolo] mask = 0,1,2 -anchors = 10,13, 16,30, 33,23, 30,61, 62,45, 59,119, 116,90, 156,198, 373,326 +anchors = 129.1513,150.1077, 104.9181,195.1675, 145.8067,186.4670, 127.9706,231.2296, 165.9494,221.8162, 162.4676,253.0225 classes=1 num=9 jitter=.3 -- Gitblit v1.10.0