cnstream/python/samples/yolov3_detection_config.json

{
  "profiler_config" : {
    "enable_profiling" : true,
    "enable_tracing" : true
  },

  "source" : {
    "class_name" : "cnstream::DataSource",
    "next_modules" : ["detector"],
    "custom_params" : {
      "reuse_cndec_buf" : "true",
      "output_type" : "mlu",
      "decoder_type" : "mlu",
      "output_buf_number" : 32,
      "device_id" : 0
    }
  },
  "detector" : {
    "class_name" : "cnstream::Inferencer",
    "parallelism" : 1,
    "max_input_queue_size" : 20,
    "next_modules" : ["osd"],
    "custom_params" : {
      "model_path" : "../../data/models/yolov3_b4c4_argb_mlu270.cambricon",
      "func_name" : "subnet0",
      "preproc_name" : "PyPreproc",
      "custom_preproc_params" : {
        "pyclass_name" : "Yolov3Preproc"
      },
      "postproc_name" : "PyPostproc",
      "custom_postproc_params" : {
        "pyclass_name" : "Yolov3Postproc",
        "threshold" : 0.5
      },
      "model_input_pixel_format" : "ARGB32",
      "batching_timeout" : 100,
      "device_id" : 0
    }
  },
  "osd" : {
    "class_name" : "cnstream::Osd",
    "parallelism" : 1,
    "max_input_queue_size" : 20,
    "next_modules" : ["encode"],
    "custom_params" : {
      "label_path" : "../../data/models/label_map_coco.txt"
    }
  },
  "encode" : {
    "class_name" : "cnstream::Encode",
    "parallelism" : 1,
    "max_input_queue_size" : 10,
    "custom_params" : {
      "frame_rate" : 25,
      "encoder_type" : "cpu",
      "file_name" : "output.mp4",
      "device_id": 0
    }
  }
}