From 52a6c30748c0cc115e77e9301ac631dd5fd8954c Mon Sep 17 00:00:00 2001 From: Joseph Redmon <pjreddie@burninator.cs.washington.edu> Date: Thu, 23 Jun 2016 07:09:36 +0000 Subject: [PATCH] better demo --- cfg/yolo.cfg | 21 +++++---------------- 1 files changed, 5 insertions(+), 16 deletions(-) diff --git a/cfg/yolo.cfg b/cfg/yolo.cfg index ee16726..1f69856 100644 --- a/cfg/yolo.cfg +++ b/cfg/yolo.cfg @@ -1,6 +1,6 @@ [net] -batch=64 -subdivisions=2 +batch=1 +subdivisions=1 height=448 width=448 channels=3 @@ -9,18 +9,10 @@ learning_rate=0.001 policy=steps -steps=100,200,300,400,500,600,700,20000,30000 -scales=2,2,1.25,1.25,1.25,1.25,1.03,.1,.1 +steps=200,400,600,20000,30000 +scales=2.5,2,2,.1,.1 max_batches = 40000 -[crop] -crop_width=448 -crop_height=448 -flip=0 -angle=0 -saturation = 1.5 -exposure = 1.5 - [convolutional] filters=64 size=7 @@ -211,14 +203,11 @@ output=4096 activation=leaky -[dropout] -probability=.5 - [connected] output= 1470 activation=linear -[region] +[detection] classes=20 coords=4 rescore=1 -- Gitblit v1.10.0