派生自 Algorithm/baseDetector

Scheaven
2021-07-27 5e878ed96fe6c05b88f7978f50d73a16f4b8a65e
update
3个文件已修改
9 ■■■■■ 已修改文件
CMakeLists.txt 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
config.json 6 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/core/ari_manager.cpp 1 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
CMakeLists.txt
@@ -23,7 +23,7 @@
set(LIBS nvinfer nvparsers nvonnxparser nvinfer_plugin stdc++fs jsoncpp log4cplus)
if(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
    set(CMAKE_CUDA_ARCHITECTURES 61 75 80)
    set(CMAKE_CUDA_ARCHITECTURES 61 75 72 80)
endif(NOT DEFINED CMAKE_CUDA_ARCHITECTURES)
config.json
@@ -2,8 +2,8 @@
  "so_file_path": "/opt/vasystem/libs/Detect/libdemo.so",
  "runtime": "/opt/vasystem/libs/Detect:/usr/local/cuda-11.1/lib64:",  // 项目所用到的环境
  "param": {
    // "model_path": "/data/disk1/project/model_dump/02_yolo/03_test.trt", // para里边自己算法可能用到的参数
    "model_path": "/data/disk1/project/model_dump/02_yolo/baseDetetor_small.bin", // para里边自己算法可能用到的参数
    "type":2
    "model_path": "/data/disk1/project/model_dump/02_yolo/baseDetect-kFLOAT-batch1.engine", // para里边自己算法可能用到的参数
    //"model_path": "/data/disk1/project/model_dump/02_yolo/baseDetetor_small.bin", // para里边自己算法可能用到的参数
    "type":1
  }
}
src/core/ari_manager.cpp
@@ -9,7 +9,6 @@
{
    Config config;
    // config.net_type = COMMON;
     printf("----0-0-0-0-0-0-----444-------------%d",m_staticStruct::type);
    if(m_staticStruct::type==2)
        config.net_type = SMALL;
    else