From 9c2c220c88aff4d0bcbdd5b03b10c6d1a7db56d3 Mon Sep 17 00:00:00 2001 From: Edmond Yoo <hj3yoo@uwaterloo.ca> Date: Sun, 16 Sep 2018 03:14:14 +0000 Subject: [PATCH] Moving files from MTGCardDetector #2 --- cfg/darknet.cfg | 21 ++++++++++++++++----- 1 files changed, 16 insertions(+), 5 deletions(-) diff --git a/cfg/darknet.cfg b/cfg/darknet.cfg index a96f4d0..60b939a 100644 --- a/cfg/darknet.cfg +++ b/cfg/darknet.cfg @@ -11,9 +11,10 @@ learning_rate=0.1 policy=poly power=4 -max_batches=500000 +max_batches=1600000 [convolutional] +batch_normalize=1 filters=16 size=3 stride=1 @@ -25,6 +26,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=32 size=3 stride=1 @@ -36,6 +38,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=64 size=3 stride=1 @@ -47,6 +50,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=128 size=3 stride=1 @@ -58,6 +62,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=256 size=3 stride=1 @@ -69,6 +74,7 @@ stride=2 [convolutional] +batch_normalize=1 filters=512 size=3 stride=1 @@ -78,20 +84,25 @@ [maxpool] size=2 stride=2 +padding=1 [convolutional] +batch_normalize=1 filters=1024 size=3 stride=1 pad=1 activation=leaky -[avgpool] - -[connected] -output=1000 +[convolutional] +filters=1000 +size=1 +stride=1 +pad=1 activation=leaky +[avgpool] + [softmax] groups=1 -- Gitblit v1.10.0