@@ -340,7 +340,7 @@ void post_process(AX_ENGINE_IO_INFO_T* io_info, AX_ENGINE_IO_T* io_data, const c
340340 auto mask_proto_ptr = (float *)io_data->pOutputs [6 ].pVirAddr ;
341341 detection::get_out_bbox_mask (proposals, objects, mask_proto_ptr, 32 , 4 , nms_threshold, input_h, input_w,
342342 mat.rows , mat.cols );
343- detection::draw_objects_mask (mat, objects, CLASS_NAMES, COCO_COLORS, " yolo11_seg_out" );
343+ // detection::draw_objects_mask(mat, objects, CLASS_NAMES, COCO_COLORS, "yolo11_seg_out");
344344 } else if (model_type == " pose" ) {
345345 float * output_ptr[3 ] = {(float *)io_data->pOutputs [0 ].pVirAddr , (float *)io_data->pOutputs [1 ].pVirAddr ,
346346 (float *)io_data->pOutputs [2 ].pVirAddr };
@@ -355,7 +355,7 @@ void post_process(AX_ENGINE_IO_INFO_T* io_info, AX_ENGINE_IO_T* io_data, const c
355355 input_h, input_w, 17 , cls_num);
356356 }
357357 detection::get_out_bbox_kps (proposals, objects, nms_threshold, input_h, input_w, mat.rows , mat.cols );
358- detection::draw_keypoints (mat, objects, KPS_COLORS, LIMB_COLORS, SKELETON, " yolo11_pose_out" );
358+ // detection::draw_keypoints(mat, objects, KPS_COLORS, LIMB_COLORS, SKELETON, "yolo11_pose_out");
359359 } else if (model_type == " obb" ) {
360360 std::vector<int > strides = {8 , 16 , 32 };
361361 std::vector<detection::GridAndStride> grid_strides;
@@ -364,7 +364,7 @@ void post_process(AX_ENGINE_IO_INFO_T* io_info, AX_ENGINE_IO_T* io_data, const c
364364 detection::obb::generate_proposals_yolov8_obb_native (grid_strides, feat_ptr, prob_threshold, proposals, input_w,
365365 input_h, cls_num);
366366 detection::obb::get_out_obb_bbox (proposals, objects, nms_threshold, input_h, input_w, mat.rows , mat.cols );
367- detection::obb::draw_objects_obb (mat, objects, OBB_CLASS_NAMES, " yolo11_obb_out" , 1 );
367+ // detection::obb::draw_objects_obb(mat, objects, OBB_CLASS_NAMES, "yolo11_obb_out", 1);
368368 }
369369}
370370
0 commit comments