From 29065df908aee41ab673cfc78a0d657e3063d5ef Mon Sep 17 00:00:00 2001
From: pansen <pansen626@sina.com>
Date: 星期六, 12 一月 2019 15:30:49 +0800
Subject: [PATCH] Merge branch 'ywv1.2_yolodebug'
---
QiaoJiaSystem/YoloServer/main.cpp | 21 ++++++++++++++++-----
1 files changed, 16 insertions(+), 5 deletions(-)
diff --git a/QiaoJiaSystem/YoloServer/main.cpp b/QiaoJiaSystem/YoloServer/main.cpp
index e223edd..438ff3e 100644
--- a/QiaoJiaSystem/YoloServer/main.cpp
+++ b/QiaoJiaSystem/YoloServer/main.cpp
@@ -23,13 +23,24 @@
// 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 = atoi(argv[1]);
+ int poolNum = atoi(argv[2]);
+ int portNum = atoi(argv[3]);
+
+ appPref.setIntData("gpu.index", gpuindex);
+ appPref.setFloatData("thresh.detect", 0.7);
+ appPref.setIntData("poolNum", poolNum);
+
+ IceRpcServer<YoloDetectServerI> server("yoloServer", portNum, "tcp");
server.setMessageSizeMax(1024 * 1024 * 50);
- server.setPoolInitSize(5);
- server.setPoolMaxSize(5);
+ server.setPoolInitSize(poolNum);
+ server.setPoolMaxSize(poolNum);
server.runWaitShutDown();
return 0;
}
--
Gitblit v1.8.0