OrgLion-ML
/
dlib
mirror of https://github.com/davisking/dlib.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Fix class index problem

This commit is contained in:
Juha Reunanen 2019-10-29 10:15:34 +02:00
parent e996bbf35f
commit 7a87f648a9
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
examples/dnn_instance_segmentation_train_ex.cpp
View File

@ -107,8 +107,7 @@ std::vector<truth_instance> rgb_label_images_to_truth_instances(
continue; continue;
const auto rgb_class_label = class_label_image(r, c); const auto rgb_class_label = class_label_image(r, c);
const auto class_label_index = rgb_label_to_index_label(rgb_class_label); const Voc2012class& voc2012_class = find_voc2012_class(rgb_class_label);
const Voc2012class& voc2012_class = classes[class_label_index];
const auto i = result_map.find(rgb_instance_label); const auto i = result_map.find(rgb_instance_label);
if (i == result_map.end()) if (i == result_map.end())