From 0c5fce0e3bcd4780c05e74e71fe1c33e6480ffca Mon Sep 17 00:00:00 2001
From: chenshijun <chenshijun@aiotlink.com>
Date: 星期四, 21 三月 2019 16:34:57 +0800
Subject: [PATCH] Yolo内存泄漏
---
QiaoJiaSystem/YoloServer/main.cpp | 32 +++++++++++++++++++++++++++-----
1 files changed, 27 insertions(+), 5 deletions(-)
diff --git a/QiaoJiaSystem/YoloServer/main.cpp b/QiaoJiaSystem/YoloServer/main.cpp
index e223edd..ff7aac0 100644
--- a/QiaoJiaSystem/YoloServer/main.cpp
+++ b/QiaoJiaSystem/YoloServer/main.cpp
@@ -7,6 +7,7 @@
#include <basic/util/app/AppConfig.h>
int main(int argc, char **argv) {
+ std::cout<<__DATE__<<" "<<__TIME__<<std::endl;
SAVE_APP_ARGS;
ENABLEGLOG(GET_STR_CONFIG("logPath").c_str());
auto ich = Ice::initialize(argc, argv);
@@ -23,13 +24,34 @@
// adapter->add(new YoloDetectServerI(), Ice::stringToIdentity(identity));
// adapter->activate();
// ich->waitForShutdown();
- appPref.setIntData("gpu.index", 0);
- appPref.setFloatData("thresh.detect", 0.7);
- IceRpcServer<YoloDetectServerI> server("yoloServer", 10003, "tcp");
+// if (argc < 3) {
+// ERR("argc error");
+// return -1;
+// }
+
+// int gpuindex = 0;
+// int poolNum = 1;
+// int portNum = 10003;
+//
+// appPref.setIntData("gpu.index", gpuindex);
+// appPref.setFloatData("thresh.detect", 0.85);
+// appPref.setIntData("poolNum", poolNum);
+
+ if (argc < 4) {
+ assert("t_value.size()");
+ }
+ appPref.setIntData("gpu.index", atoi(argv[1]));
+ appPref.setIntData("RpcYOLOPort", atoi(argv[2]));
+ appPref.setFloatData("thresh.detect", atof(argv[3]));
+
+// appPref.setIntData("poolNum", 1);//YOlO涓嶈兘澶氱嚎绋嬶紝瀹规槗宕╂簝
+
+
+ IceRpcServer<YoloDetectServerI> server("yoloServer", appPref.getIntData("RpcYOLOPort"), "tcp");
server.setMessageSizeMax(1024 * 1024 * 50);
- server.setPoolInitSize(5);
- server.setPoolMaxSize(5);
+ server.setPoolInitSize(1);//YOlO涓嶈兘澶氱嚎绋嬶紝瀹规槗宕╂簝
+ server.setPoolMaxSize(1);//YOlO涓嶈兘澶氱嚎绋嬶紝瀹规槗宕╂簝
server.runWaitShutDown();
return 0;
}
--
Gitblit v1.8.0