| | |
| | | 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) |
| | | |
| | | |
| | |
| | | "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
|
| | | }
|
| | | }
|
| | |
| | | {
|
| | | 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
|