[segmentation] add static casts

alicevision · Jun 28, 2023 · 65abaf5 · 65abaf5
1 parent 4ca0496
commit 65abaf5
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/src/aliceVision/segmentation/segmentation.cpp b/src/aliceVision/segmentation/segmentation.cpp
@@ -226,7 +226,7 @@ bool Segmentation::labelsFromModelOutput(image::Image<ScoredLabel> & labels, con
                 }
             }
 
-            labels(outputY, outputX) = {maxClasse, maxVal};
+            labels(outputY, outputX) = {static_cast<IndexT>(maxClasse), static_cast<float>(maxVal)};
         }
     }
 
@@ -239,8 +239,10 @@ bool Segmentation::processTile(image::Image<ScoredLabel> & labels, const image::
 
     std::vector<const char*> inputNames{"input"};
     std::vector<const char*> outputNames{"output"};
-    std::vector<int64_t> inputDimensions = {1, 3, _parameters.modelHeight, _parameters.modelWidth};
-    std::vector<int64_t> outputDimensions = {1, _parameters.classes.size(), _parameters.modelHeight, _parameters.modelWidth};
+    std::vector<int64_t> inputDimensions =
+        {1, 3, _parameters.modelHeight, _parameters.modelWidth};
+    std::vector<int64_t> outputDimensions =
+        {1, static_cast<int64_t>(_parameters.classes.size()), _parameters.modelHeight, _parameters.modelWidth};
 
     std::vector<float> output(_parameters.classes.size() * _parameters.modelHeight * _parameters.modelWidth);
     Ort::Value outputTensors = Ort::Value::CreateTensor<float>(
@@ -284,8 +286,10 @@ bool Segmentation::processTileGPU(image::Image<ScoredLabel> & labels, const imag
 
     std::vector<const char*> inputNames{"input"};
     std::vector<const char*> outputNames{"output"};
-    std::vector<int64_t> inputDimensions = {1, 3, _parameters.modelHeight, _parameters.modelWidth};
-    std::vector<int64_t> outputDimensions = {1, _parameters.classes.size(), _parameters.modelHeight, _parameters.modelWidth};
+    std::vector<int64_t> inputDimensions =
+        {1, 3, _parameters.modelHeight, _parameters.modelWidth};
+    std::vector<int64_t> outputDimensions =
+        {1, static_cast<int64_t>(_parameters.classes.size()), _parameters.modelHeight, _parameters.modelWidth};
 
 
     Ort::Value outputTensors = Ort::Value::CreateTensor<float>(