From 5e878ed96fe6c05b88f7978f50d73a16f4b8a65e Mon Sep 17 00:00:00 2001
From: Scheaven <xuepengqiang>
Date: 星期二, 27 七月 2021 17:11:20 +0800
Subject: [PATCH] update
---
config.json | 6 +++---
src/core/ari_manager.cpp | 1 -
CMakeLists.txt | 2 +-
3 files changed, 4 insertions(+), 5 deletions(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index 3f92f33..f195a47 100644
--- a/CMakeLists.txt
+++ b/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)
diff --git a/config.json b/config.json
index 0c210af..66558c1 100644
--- a/config.json
+++ b/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
}
}
diff --git a/src/core/ari_manager.cpp b/src/core/ari_manager.cpp
index 908ccee..d904910 100644
--- a/src/core/ari_manager.cpp
+++ b/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
--
Gitblit v1.8.0