OrgLion-ML
/
darknet
mirror of https://github.com/AlexeyAB/darknet.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

get_coco_dataset.sh downloads MS COCO dataset from images.cocodataset.org

This commit is contained in:
AlexeyAB 2018-12-21 23:07:37 +03:00
parent 49189c22e8
commit 7d0efaa936
3 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
build/darknet/x64/yolov3-voc.cfg
View File

@ -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

7
scripts/get_coco_dataset.sh
View File

@ -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

2
src/maxpool_layer.h
View File

@ -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