Hu Chunming / vpt_ascend

  #include "../ai_platform/stl_aiplatform.h"
  #include <chrono>
  
  using namespace std;
  
  
  void set_task_params(task_param &tparam, const unsigned &idx, const algorithm_type_t &algor_type) {
    auto algor_init_params = new algor_init_config_param_t;
    switch (algor_type) {
    case algorithm_type_t::FACE_SNAPSHOT: {
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
  
        basic_params->result_folder = "res/face";
        ;
        basic_params->result_folder_little = "res/face_little";
        ;
      }
      auto algor_params = new algor_config_param_snapshot;
      { algor_params->threshold = 0.5f; }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::HUMAN_SNAPSHOT: {
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
  
        basic_params->result_folder = "res/human";
        basic_params->result_folder_little = "res/human_little";
      }
      auto algor_params = new algor_config_param_snapshot;
      { algor_params->threshold = 0.5f; }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::PEDESTRIAN_FALL: {
      auto algor_params = new algor_config_param_pedestrian_fall;
      { algor_params->threshold = 0.7f; }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
  
        basic_params->result_folder = "res/fall";
        basic_params->result_folder_little = "res/fall_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::PEDESTRIAN_FIGHT: {
      auto algor_params = new algor_config_param_pedestrian_fight;
      {
        algor_params->threshold = 0.7f;
        algor_params->iou_threshold = 0.1f;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
  
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/fight";
        basic_params->result_folder_little = "res/fight_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::HUMAN_GATHER: {
  
      auto algor_params = new algor_config_param_human_gather;
      {
        algor_params->frame_stride = 1;
        // algor_params->human_count_threshold = 3;
        algor_params->human_count_threshold = 1;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/gather";
        basic_params->result_folder_little = "res/gather_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::NO_REFLECTIVE_CLOTHING: {
      auto algor_params = new algor_config_param_no_reflective_clothing;
      {
        algor_params->conf_threshold = 0.3f;
        algor_params->m = 10;
        algor_params->n = 6;
        algor_params->pedestrian_confidence_threshold = 0.3f;
        algor_params->pedestrian_min_height = 0;
        algor_params->pedestrian_min_width = 0;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/no_reflective_clothing";
        basic_params->result_folder_little = "res/no_reflective_clothing_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::NO_SAFETY_HELMET: {
      auto algor_params = new algor_config_param_no_safety_helmet;
      {
        algor_params->conf_threshold = 0.3f;
        algor_params->m = 10;
        algor_params->n = 6;
        algor_params->pedestrian_confidence_threshold = 0.3f;
        algor_params->pedestrian_min_height = 0;
        algor_params->pedestrian_min_width = 0;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/no_safety_helmet";
        basic_params->result_folder_little = "res/no_safety_helmet_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::CALL_PHONE_DET: {
      auto algor_params = new algor_config_param_call_phone;
      {
        algor_params->conf_threshold = 0.3f;
        algor_params->m = 2;
        algor_params->n = 1;
        algor_params->pedestrian_confidence_threshold = 0.1f;
        algor_params->pedestrian_min_height = 0;
        algor_params->pedestrian_min_width = 0;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/call_phone";
        basic_params->result_folder_little = "res/call_phone_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::SMOKING_DET: {
      auto algor_params = new algor_config_param_smoking;
      {
        algor_params->conf_threshold = 0.3f;
        algor_params->m = 10;
        algor_params->n = 1;
        algor_params->pedestrian_confidence_threshold = 0.3f;
        algor_params->pedestrian_min_height = 0;
        algor_params->pedestrian_min_width = 0;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/smoking";
        basic_params->result_folder_little = "res/smoking_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::VEHICLE_SNAPSHOT: {
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/vehicle";
        basic_params->result_folder_little = "res/vehicle_little";
      }
  
      auto algor_params = new algor_config_param_snapshot;
      algor_params->threshold = 0.5f; 
      algor_params->snap_frame_interval = 5;
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::NONMOTOR_VEHICLE_SNAPSHOT: {
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/nonmotor";
        basic_params->result_folder_little = "res/nonmotor_little";
      }
  
      auto algor_params = new algor_config_param_snapshot;
      { algor_params->threshold = 0.5f; }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::TAKEAWAY_MEMBER_CLASSIFICATION: {
      auto algor_params = new algor_config_param_takeaway_member_classification;
      {
        algor_params->m = 10;
        algor_params->n = 2;
        algor_params->threshold = 0.7f;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/takeaway";
        basic_params->result_folder_little = "res/takeaway_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    case algorithm_type_t::PEDESTRIAN_RETROGRADE: {
      // 578 1300 600
      auto algor_params = new algor_config_param_pedestrian_retrograde;
      {
        algor_params->conf_threshold = 0.5f;
        algor_params->minmum_height = 10;
        algor_params->minmum_width = 10;
        algor_params->direction = 0;
  
        algor_params->px1 = 578;
        algor_params->py1 = 600;
  
        algor_params->px2 = 1300;
        algor_params->py2 = 600;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/pedestrian_retrograde";
        basic_params->result_folder_little = "res/pedestrian_retrograde_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
  
    } break;
  
    case algorithm_type_t::VEHICLE_RETROGRADE: {
      // 578 1300 600
      auto algor_params = new algor_config_param_pedestrian_retrograde;
      {
        algor_params->conf_threshold = 0.5f;
        algor_params->minmum_height = 10;
        algor_params->minmum_width = 10;
        algor_params->direction = 0;
  
        algor_params->px1 = 578;
        algor_params->py1 = 600;
  
        algor_params->px2 = 1300;
        algor_params->py2 = 600;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
  
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/vehicle_retrograde";
        basic_params->result_folder_little = "res/vehicle_retrograde_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
  
    } break;
  
    case algorithm_type_t::PEDESTRIAN_TRESPASS: {
      // 578 1300 600
      auto algor_params = new algor_config_param_pedestrian_trespass;
      {
        algor_params->conf_threshold = 0.5f;
        algor_params->minmum_height = 64;
        algor_params->minmum_width = 32;
        algor_params->points_count = 4;
  
        algor_params->points[0].x_ = 200;
        algor_params->points[0].y_ = 200;
  
        algor_params->points[1].x_ = 600;
        algor_params->points[1].y_ = 200;
  
        algor_params->points[2].x_ = 600;
        algor_params->points[2].y_ = 500;
  
        algor_params->points[3].x_ = 200;
        algor_params->points[3].y_ = 500;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/pedestrian_trespass";
        basic_params->result_folder_little = "res/pedestrian_trespass_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
  
    } break;
  
    case algorithm_type_t::VEHICLE_TRESPASS: {
      // 578 1300 600
      auto algor_params = new algor_config_param_vehicle_trespass;
      {
        algor_params->conf_threshold = 0.5f;
        algor_params->minmum_height = 64;
        algor_params->minmum_width = 64;
        algor_params->points_count = 4;
  
        algor_params->points[0].x_ = 500;
        algor_params->points[0].y_ = 500;
  
        algor_params->points[1].x_ = 1500;
        algor_params->points[1].y_ = 500;
  
        algor_params->points[2].x_ = 1500;
        algor_params->points[2].y_ = 900;
  
        algor_params->points[3].x_ = 500;
        algor_params->points[3].y_ = 900;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        basic_params->algor_valid_rect.width_ = 1920;
        basic_params->algor_valid_rect.height_ = 1080;
        basic_params->result_folder = "res/vehicle_trespass";
        basic_params->result_folder_little = "res/vehicle_trespass_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
  
    } break;
  
    case algorithm_type_t::VIDEO_SNAPSHOT: {
      auto basic_params = new algor_basic_config_param_t;
      { basic_params->result_folder = "res/video_snapshot"; }
      algor_init_params->basic_param = basic_params;
    } break;
  
   
    case algorithm_type_t::ROAD_WORK_DET: {
  
      auto algor_params = new algor_config_param_road_work;
      {
        algor_params->frame_stride = 5;
        algor_params->rblock_count_threshold = 3;
        // algor_params->frame_stride = 1;
        // algor_params->rblock_count_threshold = 1;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->algor_valid_rect.top_ = 0;
        basic_params->algor_valid_rect.left_ = 0;
        // basic_params->algor_valid_rect.width_ = 1920;
        // basic_params->algor_valid_rect.height_ = 1080;
        basic_params->algor_valid_rect.width_ = 2560;
        basic_params->algor_valid_rect.height_ = 1440;
        basic_params->result_folder = "res/road_work";
        basic_params->result_folder_little = "res/road_work_little";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
     case algorithm_type_t::VIDEO_TIMING_SNAPSHOT: {
  
      auto algor_params = new algor_config_param_road_work;
      {
        algor_params->frame_stride = 5;
      }
  
      auto basic_params = new algor_basic_config_param_t;
      {
        basic_params->result_folder = "res/video_timing_snapshot";
      }
  
      algor_init_params->algor_param = algor_params;
      algor_init_params->basic_param = basic_params;
    } break;
  
    default: {
      if (algor_init_params != nullptr) {
        delete algor_init_params;
        algor_init_params = nullptr;
      }
      return;
    } break;
    }
  
    tparam.algor_config_params[idx].algor_type = algor_type;
    tparam.algor_config_params[idx].algor_init_config_param = algor_init_params;
  }
  
  static long long get_cur_time(){
      chrono::time_point<chrono::system_clock, chrono::milliseconds> tpMicro
          = chrono::time_point_cast<chrono::milliseconds>(chrono::system_clock::now());
      return tpMicro.time_since_epoch().count();
  }
  
  string createTask(void *handle, std::vector<algorithm_type_t> algor_vec, int gi){
    task_param tparam;
    // tparam.ipc_url = "rtsp://admin:ad123456@192.168.60.108:554/cam/realmonitor?channel=1&subtype=0";

    // tparam.ipc_url = "/data1/cmhu/data/bayue.mp4";
    tparam.ipc_url = "/data1/cmhu/data/Street.uvf";
    // tparam.ipc_url = "rtsp://admin:ad123456@192.168.60.165:554/cam/realmonitor?channel=1&subtype=0";

    tparam.algor_counts = algor_vec.size();
    tparam.dec_type = 2;
  
    std::string task_id_str = "test_task_id_" + std::to_string(gi);
    tparam.task_id = task_id_str.c_str();
  
    tparam.algor_config_params = new algor_config_param[tparam.algor_counts];
  
    for (size_t idx = 0; idx < algor_vec.size(); ++idx)
      set_task_params(tparam, idx, algor_vec.at(idx));
  
    const int result_code = add_task(handle, tparam);
    if (result_code != 0)
      printf("[Error]: ");
    printf("--- task_id: %s result code: %d\n", tparam.task_id, result_code);
    return task_id_str;
  }
  

  void test_gpu(int gpuID){

      tsl_aiplatform_param vptParam;
      vptParam.gpuid = gpuID;
      vptParam.trt_serialize_file = "";
  
      vptParam.log_days = 1;
      vptParam.log_level = AI_LOG_LEVEL_TRACE;
      // vptParam.log_level = AI_LOG_LEVEL_DEBUG;
      vptParam.log_mem = 64 * 1024 * 1024; // 64MB.
      vptParam.log_path = "logs/main.log";
      vptParam.vpt_thred = 0.45;
      vptParam.rblock_thred = 0.4;
  
      void *handle;
      int flag = tsl_aiplatform_init(&handle, vptParam);
      if (0 != flag) {
          printf("Init Failed! Error Code: %d\n", flag);

          return;

      } else {
          printf("Init Success\n");
      }
  

      std::vector<algorithm_type_t> algor_vec = {algorithm_type_t::FACE_SNAPSHOT, algorithm_type_t::HUMAN_SNAPSHOT,algorithm_type_t::ROAD_WORK_DET, algorithm_type_t::PEDESTRIAN_RETROGRADE, algorithm_type_t::VEHICLE_RETROGRADE,

                                              algorithm_type_t::PEDESTRIAN_TRESPASS, algorithm_type_t::VEHICLE_TRESPASS, algorithm_type_t::VEHICLE_SNAPSHOT, algorithm_type_t::NONMOTOR_VEHICLE_SNAPSHOT, algorithm_type_t::VIDEO_TIMING_SNAPSHOT};

      string task_id = createTask(handle, algor_vec, 0 + gpuID * 10);  

      // string task_id1 = createTask(handle, algor_vec, 1 + gpuID * 10);  
      // string task_id2 = createTask(handle, algor_vec, 2 + gpuID * 10);  

      while (getchar() != 'q');
      
      finish_task(handle, (char*)task_id.data(), 0);

      tsl_aiplatform_release(&handle);

  }
  
  int main(int argc, char *argv[]) {
      printf("new test\n");
  
      if (argc < 4) {
      fprintf(stderr, "./xxx 0 2 10 1 ## [start_ai_id, end_ai_id) repeat_num gpu_id\n");
      return -1;
      }
  
      //! load params.
      int start_id = atoi(argv[1]);
      int end_id = atoi(argv[2]);
      int repeat_num = atoi(argv[3]);
      int gpuID = atoi(argv[4]);
  
      test_gpu(0);

      // test_gpu(1);
      // test_gpu(2);
      // test_gpu(3);

  
      while (getchar() != 'q');

      printf("Done.\n"); 

  
      return 0;
  }