瀏覽代碼

SemSegTools: added 256+ label support

Sven Sickert 9 年之前
父節點
當前提交
b1f7255f01
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      semseg/SemSegTools.cpp

+ 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() );