[update] update depth_anything.

Author: Abandon-ht

projects/llm_framework/main_depth_anything/src/main.cpp

@@ -160,7 +160,7 @@ class llm_task {
         cv::Mat camera_data(mode_config_.img_h, mode_config_.img_w, CV_8UC2, (void *)msg.data());
         cv::Mat rgb;
         cv::cvtColor(camera_data, rgb, cv::COLOR_YUV2RGB_YUYV);
-        return inference(rgb, false);
+        return inference(rgb, true);
     }
 
     bool inference_raw_rgb(const std::string &msg)
@@ -195,13 +195,14 @@ class llm_task {
             int ret = -1;
             std::vector<uint8_t> image(mode_config_.img_w * mode_config_.img_h * 3, 0);
             common::get_input_data_no_letterbox(src, image, mode_config_.img_h, mode_config_.img_w, bgr2rgb);
+            cv::Mat img_mat(mode_config_.img_h, mode_config_.img_w, CV_8UC3, image.data());
             depth_anything_->SetInput((void *)image.data(), 0);
             if (0 != depth_anything_->RunSync()) {
                 SLOGE("Run depth_anything model failed!\n");
                 throw std::string("depth_anything_ RunSync error");
             }
             std::vector<detection::Object> objects;
-            depth_anything_->Post_Process(src, mode_config_.model_type);
+            depth_anything_->Post_Process(img_mat, mode_config_.model_type);
             std::vector<nlohmann::json> depth_anything_output;
             if (out_callback_) out_callback_(depth_anything_output, true);
         } catch (...) {

projects/llm_framework/main_yolo/src/main.cpp

@@ -199,13 +199,14 @@ class llm_task {
             int ret = -1;
             std::vector<uint8_t> image(mode_config_.img_w * mode_config_.img_h * 3, 0);
             common::get_input_data_letterbox(src, image, mode_config_.img_h, mode_config_.img_w, bgr2rgb);
+            cv::Mat img_mat(mode_config_.img_h, mode_config_.img_w, CV_8UC3, image.data());
             yolo_->SetInput((void *)image.data(), 0);
             if (0 != yolo_->RunSync()) {
                 SLOGE("Run yolo model failed!\n");
                 throw std::string("yolo_ RunSync error");
             }
             std::vector<detection::Object> objects;
-            yolo_->Post_Process(src, mode_config_.img_w, mode_config_.img_h, mode_config_.cls_num,
+            yolo_->Post_Process(img_mat, mode_config_.img_w, mode_config_.img_h, mode_config_.cls_num,
                                 mode_config_.pron_threshold, mode_config_.nms_threshold, objects,
                                 mode_config_.model_type);
             std::vector<nlohmann::json> yolo_output;