Эх сурвалжийг харах

SemSegTools: added 256+ label support

Sven Sickert 9 жил өмнө
parent
commit
b1f7255f01

+ 1 - 1
semseg/SemSegTools.cpp

@@ -283,7 +283,7 @@ void SemSegTools::collectTrainingExamples (
         std::cerr << "SemSegTools: Collecting pixel examples from localization info: "
                   << imgfn << std::endl;
 
-        NICE::Image pixelLabels (xsize, ysize);
+        NICE::ImageT<int> pixelLabels (xsize, ysize);
         pixelLabels.set(0);
         locResult->calcLabeledImage ( pixelLabels, cn.getBackgroundClass() );