xavier
808a84131c
The new test works both with TensorRT and cuDNN. Preprocessing and Postprocessing are missing. Add ClippedReLU (for ReLU6), groups for Conv2d, additional bias for convolution. Other minors: -move the timer in the detector to measure all the processing time for a given frame (both centernet and yolo); -add int8 flag. Signed-off-by: Micaela Verucchi <micaelaverucchi@gmail.com> Davide Sapienza <sapienza.dav@gmail.com>
7 lines
361 B
YAML
7 lines
361 B
YAML
classes : 80 #number of classes
|
|
map_points : 101 #number of recall points (0 for all, 101 for COCO, 11 PascalVOC)
|
|
map_levels : 10 #number of IoU step for the AP
|
|
map_step : 0.05 #step of IoU
|
|
IoU_thresh : 0.5 #starting IoU threshold
|
|
conf_thresh : 0.0 #threshold on the condifence of the bbox
|
|
verbose : false #print on screen information |