Change README.md, add config.yaml, refactoring

Signed-off-by: xavier <micaelaverucchi@gmail.com>
2020-02-11 11:25:42 +01:00
parent e01d7c6b13
commit d0026ff235
6 changed files with 73 additions and 42 deletions
@@ -7,6 +7,7 @@ this branch actually work on every NVIDIA GPU that support the dependencies:
 * CUDNN 7.603
 * TENSORRT 6.01
 * OPENCV 4.1
+* yaml-cpp 0.5.2 (sudo apt install libyaml-cpp-dev)

 ## Workflow
 The recommended workflow follow these step:
@@ -112,4 +113,23 @@ this will genereate resnet101_cnet.rt and dla34_cnet.rt file that can be used fo
 ./centernet_demo                               # launch detection on a demo video
 ./centernet_demo resnet101_cnet.rt /dev/video0 # launch detection on device 0
 ./centernet_demo dla34_cnet.rt /dev/video0     # launch detection on device 0
+```
+
+## mAP demo
+To compute mAP, precision, recall and f1score, run the map_demo.
+
+The following parameters are needed:
+```
+./map_demo <network rt> <network type [y|c]> <labels file path> <config file path>
+```
+where 
+* <network rt>: rt file of a choosen network on wich compute the mAP.
+* <network type [y|c]>: type of network. Right now only y(yolo) and c(centernet) are allowed
+* <labels file path>: path to a text file containing all the paths of the groundtruth labels. It is important that all the labels of the groundtruth are in a folder called 'labels'. In the folder containing the folder 'labels' there should be also a folder 'images', containing all the groundtruth images having the same same as the labels. To better understand, if there is a label path/to/labels/000001.txt there should be a corresponding image path/to/images/000001.jpg. 
+* <config file path>: path to a yaml file with the parameters needed for the mAP computation, similar to demo/config.yaml
+
+Example:
+
+```
+./map_demo dla34_cnet.rt c path/to/labelsCOCO_val2017.txt ../demo/config.yaml
 ```