Fixed bug of Moha

tudelft3d · Jun 4, 2023 · 9cdec84 · 9cdec84
1 parent 28846d1
commit 9cdec84
Show file tree

Hide file tree

Showing 2 changed files with 52 additions and 19 deletions.
diff --git a/semantic_urban_mesh_segmentation/mesh_io.cpp b/semantic_urban_mesh_segmentation/mesh_io.cpp
@@ -326,28 +326,61 @@ namespace semantic_mesh_segmentation
 		if (enable_augment && train_test_predict_val == 0)
 		{
 			std::cout << "	Start to read augmented features " << s1_test << std::endl;
-			str_ostemp
-				<< root_path
-				<< folder_names_level_0[1]
-				<< folder_names_level_1[4]
-				<< s1_test
-				<< prefixs[0]
-				<< prefixs[3]
-				<< prefixs[14]
-				<< ".ply";
+			if (processing_mode == 0)
+			{
+				str_ostemp
+					<< root_path
+					<< folder_names_level_0[1]
+					<< folder_names_level_1[4]
+					<< s1_test
+					<< prefixs[0]
+					<< prefixs[3]
+					<< prefixs[14]
+					<< ".ply";
+			}
+			else
+			{
+				str_ostemp
+					<< root_path
+					<< folder_names_level_0[8]
+					<< sota_folder_path
+					<< folder_names_level_0[1]
+					<< folder_names_level_1[4]
+					<< s1_test
+					<< prefixs[0]
+					<< prefixs[3]
+					<< prefixs[14]
+					<< ".ply";
+			}
+
 		}
 		else
 		{
 			std::cout << "	Start to read features " << s1_test << std::endl;
-
-			str_ostemp
-				<< root_path
-				<< folder_names_level_0[1]
-				<< folder_names_level_1[train_test_predict_val]
-				<< s1_test
-				<< prefixs[0]
-				<< prefixs[3]
-				<< ".ply";
+			if (processing_mode == 0)
+			{
+				str_ostemp
+					<< root_path
+					<< folder_names_level_0[1]
+					<< folder_names_level_1[train_test_predict_val]
+					<< s1_test
+					<< prefixs[0]
+					<< prefixs[3]
+					<< ".ply";
+			}
+			else
+			{
+				str_ostemp
+					<< root_path
+					<< folder_names_level_0[8]
+					<< sota_folder_path
+					<< folder_names_level_0[1]
+					<< folder_names_level_1[train_test_predict_val]
+					<< s1_test
+					<< prefixs[0]
+					<< prefixs[3]
+					<< ".ply";
+			}
 		}
 
 		std::string str_temp = str_ostemp.str().data();

diff --git a/semantic_urban_mesh_segmentation/operation_function.cpp b/semantic_urban_mesh_segmentation/operation_function.cpp
@@ -1051,7 +1051,7 @@ namespace semantic_mesh_segmentation
 			use_color_features[2] = false;
 			use_color_features[6] = false;
 
-			//run(operating_mode::Mesh_feature_extraction);
+			run(operating_mode::Mesh_feature_extraction);
 			run(operating_mode::Train_and_Test_config);
 			run(operating_mode::Data_evaluation_for_all_tiles_config);
 			break;