SSD: custom tlt training result in AP:0 for all my classes

Accelerated Computing Intelligent Video Analytics TAO Toolkit
,
6

In your spec as below, please set to lowercase class name. In your tfrecord, it is already set to lowercase.

target_class_mapping {
key: “BLACK_ARTENGO”
value: “BLACK_ARTENGO”
}
target_class_mapping {
key: “BLUE_ARTENGO”
value: “BLUE_ARTENGO”
}
target_class_mapping {
key: “RED_KUIKMA”
value: “RED_KUIKMA”
}

Reference: Mean average_precision is 0% for all classes using Detectnet_V2 - #2 by Morganh