浏览代码

adapted testSemSeg such that results are written with proper names

Alexander Freytag 12 年之前
父节点
当前提交
b556102185
共有 1 个文件被更改,包括 17 次插入8 次删除
  1. 17 8
      progs/testSemanticSegmentation.cpp

+ 17 - 8
progs/testSemanticSegmentation.cpp

@@ -202,14 +202,23 @@ int main( int argc, char **argv )
 
       if ( write_results )
       {
-        char filename[1024];
-        char *format = ( char * )"ppm";
-        sprintf( filename, "%06d.%s", fileno, format );
-        std::string origfilename = resultdir + "/orig_" + string( filename );
-        cerr << "Writing to file " << origfilename << endl;
-        orig.write( origfilename );
-        rgb.write( resultdir + "/result_" + string( filename ) );
-        rgb_gt.write( resultdir + "/groundtruth_" + string( filename ) );
+//         char filename[1024];
+//         char *format = ( char * )"ppm";
+//         sprintf( filename, "%06d.%s", fileno, format );
+//         std::string origfilename = resultdir + "/orig_" + string( filename );
+//         cerr << "Writing to file " << origfilename << endl;
+//         orig.write( origfilename );
+//         rgb.write( resultdir + "/result_" + string( filename ) );
+//         rgb_gt.write( resultdir + "/groundtruth_" + string( filename ) );
+        
+        std::stringstream out;       
+        std::vector< std::string > myList;
+        StringTools::split ( Globals::getCurrentImgFN (), '/', myList );
+        out << resultdir << "/" << myList.back();
+        cerr << "Writing to file " << resultdir << "/"<< myList.back() << endl;
+        orig.write ( out.str() + "_orig.ppm" );
+        rgb.write ( out.str() + "_result.ppm" );
+        rgb_gt.write ( out.str() + "_groundtruth.ppm" );
       }
 
       if ( show_result )