diff --git a/Makefile b/Makefile index 2907e9129100129e07cb4f0cdc035170a3912e56..93762edf06f1d4683100f8858c22e4fea2c98aee 100644 --- a/Makefile +++ b/Makefile @@ -1,6 +1,6 @@ GPU=1 CUDNN=1 -CUDNN_HALF=0 +CUDNN_HALF=1 OPENCV=0 AVX=0 OPENMP=0 diff --git a/backup/yolov3-voc_final.weights b/backup/yolov3-voc_final.weights new file mode 100644 index 0000000000000000000000000000000000000000..11836d94cacdaa89efd173f38447302f734842a6 Binary files /dev/null and b/backup/yolov3-voc_final.weights differ diff --git a/backup/yolov3-voc_last.weights b/backup/yolov3-voc_last.weights new file mode 100644 index 0000000000000000000000000000000000000000..11836d94cacdaa89efd173f38447302f734842a6 Binary files /dev/null and b/backup/yolov3-voc_last.weights differ diff --git a/cfg/yolov3-voc.cfg b/cfg/yolov3-voc.cfg index 5198233562383a21ec184584c507ee00926e4bc1..61ccf80ed9ae56ab848e634341236c56b0a6fa5a 100644 --- 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 diff --git a/predictions.jpg b/predictions.jpg new file mode 100644 index 0000000000000000000000000000000000000000..c0b1499917c848ffa123e4a6598c8a633e6af5b9 Binary files /dev/null and b/predictions.jpg differ