将下面的代码修改，加入展示图片和检测视频 // Copyright (c) 2023 by Rockchip Electronics Co., Ltd. All Rights Reserved. // // Licensed under the Apache License, Version 2.0 (the "License"); // you may not use this file except in compliance with the License. // You may obtain a copy of the License at // // http://www.apache.org/licenses/LICENSE-2.0 // // Unless required by applicable law or agreed to in writing, software // distributed under the License is distributed on an "AS IS" BASIS, // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. // See the License for the specific language governing permissions and // limitations under the License.

/------------------------------------------- Includes -------------------------------------------/ #include <stdint.h> #include <stdio.h> #include <stdlib.h> #include <string.h> #include<time.h> #include "yolov8_seg.h" #include "image_utils.h" #include "file_utils.h" #include "image_drawing.h" #include <opencv2/opencv.hpp> #include <opencv2/highgui/highgui.hpp>

/------------------------------------------- Main Function -------------------------------------------/ int main(int argc, char **argv) { if (argc != 3) { printf("%s <model_path> <image_path>\n", argv[0]); return -1; }

clock_t start_time, end_time;
double cpu_time_used;

const char *model_path = argv[1];
const char *image_path = argv[2];

unsigned char class_colors[][3] = {
    {255, 56, 56}
};

int ret;
rknn_app_context_t rknn_app_ctx;
memset(&rknn_app_ctx, 0, sizeof(rknn_app_context_t));

init_post_process();
start_time = clock();  // 记录开始时间


ret = init_yolov8_seg_model(model_path, &rknn_app_ctx);
if (ret != 0)
{
    printf("init_yolov8_seg_model fail! ret=%d model_path=%s\n", ret, model_path);
    goto out;
}

image_buffer_t src_image;
memset(&src_image, 0, sizeof(image_buffer_t));
ret = read_image(image_path, &src_image);

if (ret != 0)
{
    printf("read image fail! ret=%d image_path=%s\n", ret, image_path);
    goto out;
}

object_detect_result_list od_results;
ret = inference_yolov8_seg_model(&rknn_app_ctx, &src_image, &od_results);
if (ret != 0)
{
    printf("init_yolov8_seg_model fail! ret=%d\n", ret);
    goto out;
}

// draw mask
if (od_results.count >= 1)
{
    int width = src_image.width;
    int height = src_image.height;
    char *ori_img = (char *)src_image.virt_addr;
    int cls_id = od_results.results[0].cls_id;
    uint8_t *seg_mask = od_results.results_seg[0].seg_mask;
    float alpha = 0.5f; // opacity
    for (int j = 0; j < height; j++)
    {
        for (int k = 0; k < width; k++)
        {
            int pixel_offset = 3 * (j * width + k);
            if (seg_mask[j * width + k] != 0)
            {
                ori_img[pixel_offset + 0] = (unsigned char)clamp(class_colors[seg_mask[j * width + k] % N_CLASS_COLORS][0] * (1 - alpha) + ori_img[pixel_offset + 0] * alpha, 0, 255); // r
                ori_img[pixel_offset + 1] = (unsigned char)clamp(class_colors[seg_mask[j * width + k] % N_CLASS_COLORS][1] * (1 - alpha) + ori_img[pixel_offset + 1] * alpha, 0, 255); // g
                ori_img[pixel_offset + 2] = (unsigned char)clamp(class_colors[seg_mask[j * width + k] % N_CLASS_COLORS][2] * (1 - alpha) + ori_img[pixel_offset + 2] * alpha, 0, 255); // b
            }
        }
    }
    free(seg_mask);
}
end_time = clock();
cpu_time_used = ((double)(end_time-start_time)) / CLOCKS_PER_SEC;
printf("Toral time: %f seconds\n",cpu_time_used);

// draw boxes
char text[256];
for (int i = 0; i < od_results.count; i++)
{
    object_detect_result *det_result = &(od_results.results[i]);
    printf("%s @ (%d %d %d %d) %.3f\n", coco_cls_to_name(det_result->cls_id),
           det_result->box.left, det_result->box.top,
           det_result->box.right, det_result->box.bottom,
           det_result->prop);
    int x1 = det_result->box.left;
    int y1 = det_result->box.top;
    int x2 = det_result->box.right;
    int y2 = det_result->box.bottom;

    draw_rectangle(&src_image, x1, y1, x2 - x1, y2 - y1, COLOR_RED, 3);
    sprintf(text, "%s %.1f%%", coco_cls_to_name(det_result->cls_id), det_result->prop * 100);
    draw_text(&src_image, text, x1, y1 - 16, COLOR_BLUE, 10);
}

out: deinit_post_process();

ret = release_yolov8_seg_model(&rknn_app_ctx);
if (ret != 0)
{
    printf("release_yolov8_seg_model fail! ret=%d\n", ret);
}

if (src_image.virt_addr != NULL)
{
    free(src_image.virt_addr);
}

return 0;

}