Skip to content

Cast ground truth class name datatype to str before computing metrics

Najam Syed requested to merge dev/fix_class_datatype into master

@benjamin.pikus @christopher.funk

When ground truth class names are numeric or saved as integers in the labels files, mAP will incorrectly be computed as 0. This is because prediction class names are cast to str before metrics are computed but ground truth class names are not. This MR makes a simple change that ensures ground truth class names are converted str as well.

Edited by Najam Syed

Merge request reports

Loading