yolov5_nvinferserver_config.txt

################################################################################
# SPDX-FileCopyrightText: Copyright (c) 2018-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
# SPDX-License-Identifier: LicenseRef-NvidiaProprietary
#
# NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
# property and proprietary rights in and to this material, related
# documentation and any modifications thereto. Any use, reproduction,
# disclosure or distribution of this material and related documentation
# without an express license agreement from NVIDIA CORPORATION or
# its affiliates is strictly prohibited.
################################################################################

infer_config {
  unique_id: 1
  gpu_ids: [0]
  max_batch_size: 1
  backend {
    inputs: [ {
      name: "images"
    }]
    outputs: [
      {name: "output0"}
    ]
    triton {
      model_name: "YOLOV5_FP16"
      version: 1
      model_repo {
        root: "../../../sugon_samples/triton_model_repo"
        strict_model_config: true
        backend_dir: "/opt/deepstream/third_party/backends"
        log_level: 1
      }
    }
  }

  preprocess {
    network_format: IMAGE_FORMAT_RGB
    tensor_order: TENSOR_ORDER_LINEAR
    maintain_aspect_ratio: 0
    frame_scaling_hw: FRAME_SCALING_HW_DEFAULT
    frame_scaling_filter: 1
    normalize {
      scale_factor: 0.00392157
      channel_offsets: [0, 0, 0]
    }
  }

  postprocess {
    labelfile_path: "../../../sugon_samples/labels/YOLOV5/labels.txt"
    detection {
      num_detected_classes: 80
      per_class_params {
        key: 0
        value { pre_threshold: 0.4 }
      }
      nms {
        confidence_threshold:0.2
        topk:20
        iou_threshold:0.5
      }
      custom_parse_bbox_func: "parseYolov5OutputBoundingBoxFP32"
    }
  }

  extra {
    copy_input_to_host_buffers: false
    output_buffer_pool_size: 2
  }

  custom_lib {
    path: "/opt/deepstream/lib/libnvdsinferserver__customparser_sugon.so"
  }
}
input_control {
  process_mode: PROCESS_MODE_FULL_FRAME
  operate_on_gie_id: -1
  interval: 0
}