mirror/darknet
1
0
Fork 0
mirror of https://github.com/pjreddie/darknet.git synced 2023-08-10 21:13:14 +03:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
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
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] [net]
# Testing # Testing
batch=1 # batch=1
subdivisions=1 # subdivisions=1
# Training # Training
# batch=64 batch=64
# subdivisions=16 subdivisions=32
width=416 width=416
height=416 height=416
channels=3 channels=3

7
scripts/get_coco_dataset.sh
View File

@ -8,8 +8,11 @@ mkdir images
cd images cd images
# Download Images # Download Images
wget -c https://pjreddie.com/media/files/train2014.zip #very slow downloading
wget -c https://pjreddie.com/media/files/val2014.zip #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
unzip -q train2014.zip 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 #ifdef GPU
void forward_maxpool_layer_gpu(maxpool_layer l, network_state state); void forward_maxpool_layer_gpu(maxpool_layer l, network_state state);
void backward_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); void cudnn_maxpool_setup(maxpool_layer *l);
#endif // CUDNN
#endif // GPU #endif // GPU
#endif #endif