second

a67a850b · liangpanfeng · a2295ed5 · a67a850b · a67a850b · a67a850b
5 changed file
--- a/Makefile
+++ b/Makefile
 GPU=1
 CUDNN=1
-CUDNN_HALF=0
+CUDNN_HALF=1
 OPENCV=0
 AVX=0
 OPENMP=0

--- a/backup/yolov3-voc_final.weights
+++ b/backup/yolov3-voc_final.weights
--- a/backup/yolov3-voc_last.weights
+++ b/backup/yolov3-voc_last.weights
--- a/cfg/yolov3-voc.cfg
+++ b/cfg/yolov3-voc.cfg
@@ -3,10 +3,10 @@
 # batch=1
 # subdivisions=1
 # Training
-batch=1
+batch=4
 subdivisions=1
-width=416
-height=416
+width=320
+height=320
 channels=3
 momentum=0.9
 decay=0.0005
@@ -17,7 +17,7 @@ hue=.1

 learning_rate=0.001
 burn_in=1000
-max_batches = 50200
+max_batches = 500
 policy=steps
 steps=40000,45000
 scales=.1,.1

--- a/predictions.jpg
+++ b/predictions.jpg