update

Author: irexyc

demo/csrc/c/image_segmentation.cpp

@@ -59,14 +59,14 @@ int main(int argc, char* argv[]) {
 
   cv::Mat color_mask = cv::Mat::zeros(result->height, result->width, CV_8UC3);
   int pos = 0;
+  std::vector<int> idxs(result->classes);
   for (auto iter = color_mask.begin<cv::Vec3b>(); iter != color_mask.end<cv::Vec3b>(); ++iter) {
     // output mask
     if (result->mask) {
       *iter = palette[result->mask[pos++]];
     }
     // output score
     if (result->score) {
-      std::vector<int> idxs(result->classes);
       std::iota(idxs.begin(), idxs.end(), 0);
       auto k = std::max_element(
                    idxs.begin(), idxs.end(),