From 7d0efaa936c6b5925e68b2a55f42a89b931ca3d1 Mon Sep 17 00:00:00 2001 From: AlexeyAB Date: Fri, 21 Dec 2018 23:07:37 +0300 Subject: [PATCH] get_coco_dataset.sh downloads MS COCO dataset from images.cocodataset.org --- build/darknet/x64/yolov3-voc.cfg | 8 ++++---- scripts/get_coco_dataset.sh | 7 +++++-- src/maxpool_layer.h | 2 -- 3 files changed, 9 insertions(+), 8 deletions(-) diff --git a/build/darknet/x64/yolov3-voc.cfg b/build/darknet/x64/yolov3-voc.cfg index 3f3e8dfb..8f7ba54d 100644 --- a/build/darknet/x64/yolov3-voc.cfg +++ b/build/darknet/x64/yolov3-voc.cfg @@ -1,10 +1,10 @@ [net] # Testing - batch=1 - subdivisions=1 +# batch=1 +# subdivisions=1 # Training -# batch=64 -# subdivisions=16 +batch=64 +subdivisions=32 width=416 height=416 channels=3 diff --git a/scripts/get_coco_dataset.sh b/scripts/get_coco_dataset.sh index 28463015..d7911e6c 100644 --- a/scripts/get_coco_dataset.sh +++ b/scripts/get_coco_dataset.sh @@ -8,8 +8,11 @@ mkdir images cd images # Download Images -wget -c https://pjreddie.com/media/files/train2014.zip -wget -c https://pjreddie.com/media/files/val2014.zip +#very slow downloading +#wget -c https://pjreddie.com/media/files/train2014.zip +#wget -c https://pjreddie.com/media/files/val2014.zip +wget -c http://images.cocodataset.org/zips/train2014.zip +wget -c http://images.cocodataset.org/zips/val2014.zip # Unzip unzip -q train2014.zip diff --git a/src/maxpool_layer.h b/src/maxpool_layer.h index 2b6c277a..3a60cfe9 100644 --- a/src/maxpool_layer.h +++ b/src/maxpool_layer.h @@ -17,9 +17,7 @@ void backward_maxpool_layer(const maxpool_layer l, network_state state); #ifdef GPU void forward_maxpool_layer_gpu(maxpool_layer l, network_state state); void backward_maxpool_layer_gpu(maxpool_layer l, network_state state); -#ifdef CUDNN void cudnn_maxpool_setup(maxpool_layer *l); -#endif // CUDNN #endif // GPU #endif