From 7e21b10aeec4d3f302e63517ae5a5ec672a06782 Mon Sep 17 00:00:00 2001 From: xavier Date: Tue, 11 Feb 2020 14:56:54 +0100 Subject: [PATCH] Add script to download COCO val2017 Signed-off-by: xavier --- README.md | 10 ++++++++-- demo/demo/map.cpp | 4 ++-- download_validation.sh | 8 ++++++++ 3 files changed, 18 insertions(+), 4 deletions(-) create mode 100644 download_validation.sh diff --git a/README.md b/README.md index b4ab4a6..9219939 100644 --- a/README.md +++ b/README.md @@ -118,7 +118,12 @@ this will genereate resnet101_cnet.rt and dla34_cnet.rt file that can be used fo ## mAP demo To compute mAP, precision, recall and f1score, run the map_demo. -The following parameters are needed: +A validation set is needed. To download COCO_val2017 run (form the root folder): +``` +bash download_validation.sh +``` + +To compute the map, the following parameters are needed: ``` ./map_demo ``` @@ -131,5 +136,6 @@ where Example: ``` -./map_demo dla34_cnet.rt c path/to/labelsCOCO_val2017.txt ../demo/config.yaml +cd build +./map_demo dla34_cnet.rt c ../demo/COCO_val2017/all_labels.txt ../demo/config.yaml ``` \ No newline at end of file diff --git a/demo/demo/map.cpp b/demo/demo/map.cpp index bc578d8..13636fb 100644 --- a/demo/demo/map.cpp +++ b/demo/demo/map.cpp @@ -31,7 +31,7 @@ int main(int argc, char *argv[]) char ntype = 'y'; char *config_filename = "../demo/config.yaml"; char * net = "yolo3.rt"; - char * labels_path = "/media/887E650E7E64F67A/val2017/all_labels2017.txt"; + char * labels_path = "../demo/COCO_val2017/all_labels.txt"; bool show = false; bool write_dets = false; int n_images = 1000; @@ -42,7 +42,7 @@ int main(int argc, char *argv[]) ntype = argv[2][0]; if(argc > 3) labels_path = argv[3]; - if(argc > 3) + if(argc > 4) config_filename = argv[4]; tk::dnn::Yolo3Detection yolo; diff --git a/download_validation.sh b/download_validation.sh new file mode 100644 index 0000000..4e07aae --- /dev/null +++ b/download_validation.sh @@ -0,0 +1,8 @@ +#!/bin/bash + +cd demo +wget https://cloud.hipert.unimore.it/s/LNxBDk4wzqXPL8c/download -O COCO_val2017.zip +unzip -d COCO_val2017 COCO_val2017.zip +rm COCO_val2017.zip +cd COCO_val2017/ +realpath labels/* > all_labels.txt