三、Deepsrtream 保存图片

最新推荐文章于 2023-11-24 15:20:23 发布

原创

最新推荐文章于 2023-11-24 15:20:23 发布 · 1k 阅读

6 ·

CC 4.0 BY-SA版权

文章标签：

#人工智能 #深度学习

文章介绍了在DeepStream的demo中如何正确地保存图片，特别提到了优快云上常见示例的内存泄漏问题。通过使用GstBuffer，NvBufSurface和CUDA进行图像处理和转换，然后利用OpenCV将NV12/RGBA格式转换为BGR并保存为JPEG文件，同时确保了内存管理的正确性。

前言

DeepStream提供的demo中有相关保存图片的示例，等有时间再进行总结。优快云有许多关于DeepStream保存图片的文章，大部分都存在内存泄漏问题。

参考

#if 1
  static int dump = 0;
  if (dump < 150) {
   
   
    GstMapInfo in_map_info;
    NvBufSurface *surface = NULL;

    memset (&in_map_info, 0, sizeof (in_map_info));
    if (!gst_buffer_map (buf, &in_map_info, GST_MAP_READ)) {
   
   
      g_print ("Error: Failed to map gst buffer\n");
      gst_buffer_unmap (buf, &in_map_info);
      return GST_PAD_PROBE_OK;
    }
    cudaError_t cuda_err;

    NvBufSurfTransformRect src_rect, dst_rect;
    surface = (NvBufSurface *) in_map_info.data;  
  
    int batch_size= surface->batchSize;
    printf("\nBatch Size : %d, resolution : %dx%d \n",batch_size,
        surface->surfaceList