소스 검색

bug fixes

Bjoern Froehlich 13 년 전
부모
커밋
d383c40a76
1개의 변경된 파일8개의 추가작업 그리고 4개의 파일을 삭제
  1. 8 4
      semseg/SemSegContextTree.cpp

+ 8 - 4
semseg/SemSegContextTree.cpp

@@ -5,6 +5,7 @@
 
 #include "vislearning/cbaselib/CachedExample.h"
 #include "vislearning/cbaselib/PascalResults.h"
+#include "vislearning/baselib/ColorSpace.h"
 #include "objrec/segmentation/RSMeanShift.h"
 #include "objrec/segmentation/RSGraphBased.h"
 #include "core/basics/numerictools.h"
@@ -717,7 +718,7 @@ double SemSegContextTree::getBestSplit( std::vector<NICE::MultiChannelImageT<dou
 
   if ( featcounter < minFeats )
   {
-    cout << "only " << featcounter << " feats in current node -> it's a leaf" << endl;
+    //cout << "only " << featcounter << " feats in current node -> it's a leaf" << endl;
     return 0.0;
   }
 
@@ -784,7 +785,7 @@ double SemSegContextTree::getBestSplit( std::vector<NICE::MultiChannelImageT<dou
 
   if ( globent < 0.5 )
   {
-    cout << "globent to small: " << globent << endl;
+    //cout << "globent to small: " << globent << endl;
     return 0.0;
   }
 
@@ -1092,7 +1093,8 @@ void SemSegContextTree::train( const MultiDataset *md )
 
   int imgcounter = 0;
 
-  /*MultiChannelImageT<int> ttmp2(0,0,0);
+  /*
+  MultiChannelImageT<int> ttmp2(0,0,0);
   MultiChannelImageT<double> ttmp1(100,100,1);
   MultiChannelImageT<double> tint(100,100,1);
   ttmp1.setAll(1.0);
@@ -1178,6 +1180,7 @@ void SemSegContextTree::train( const MultiDataset *md )
       }
     }
 
+    allfeats[imgcounter] = ColorSpace::rgbtolab(allfeats[imgcounter]);
 #endif
 
     // getting groundtruth
@@ -1546,7 +1549,8 @@ void SemSegContextTree::semanticseg( CachedExample *ce, NICE::Image & segresult,
       }
     }
   }
-
+  
+  feats = ColorSpace::rgbtolab(feats);
 #endif
 
   bool allleaf = false;