vehicle_cts_mlu270.json 1.2 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. {
  2. // vehicle color type sid recognization
  3. "detector" : {
  4. "class_name" : "cnstream::Inferencer",
  5. "parallelism" : 1,
  6. "max_input_queue_size" : 20,
  7. "next_modules" : ["vehicle_cts"],
  8. "custom_params" : {
  9. "model_path" : "../../../data/models/resnet34_ssd_b16c16_mlu270.cambricon",
  10. "func_name" : "subnet0",
  11. "postproc_name" : "PostprocSsd",
  12. "threshold" : 0.5,
  13. "model_input_pixel_format" : "BGRA32",
  14. "device_id" : 0
  15. }
  16. },
  17. "vehicle_cts" : {
  18. "class_name" : "cnstream::Inferencer",
  19. "parallelism" : 1,
  20. "max_input_queue_size" : 20,
  21. "next_modules" : ["osd"],
  22. "custom_params" : {
  23. "model_path" : "../../../data/models/vehicle_cts_b4c4_bgra_mlu270.cambricon",
  24. "func_name" : "subnet0",
  25. "object_infer" : true, // input with detection objects
  26. "obj_filter_name" : "CarFilter", // filter objects expect vehicles
  27. "postproc_name" : "PostprocVehicleCts",
  28. "model_input_pixel_format" : "BGRA32",
  29. "device_id" : 0
  30. }
  31. },
  32. "osd" : {
  33. "class_name" : "cnstream::Osd",
  34. "parallelism" : 1,
  35. "max_input_queue_size" : 20,
  36. "custom_params" : {
  37. "label_path" : "../../../data/models/label_voc.txt",
  38. "attr_keys" : "COLOR, TYPE, TOWARDS"
  39. }
  40. }
  41. }