Pārlūkot izejas kodu

remove batch_size arg

jack 5 gadi atpakaļ
vecāks
revīzija
203c90ed54

+ 1 - 2
deploy/cpp/demo/classifier.cpp

@@ -57,8 +57,7 @@ int main(int argc, char** argv) {
              FLAGS_use_gpu,
              FLAGS_use_trt,
              FLAGS_gpu_id,
-             FLAGS_key,
-             FLAGS_batch_size);
+             FLAGS_key);
 
   // 进行预测
   double total_running_time_s = 0.0;

+ 1 - 2
deploy/cpp/demo/detector.cpp

@@ -62,8 +62,7 @@ int main(int argc, char** argv) {
              FLAGS_use_gpu,
              FLAGS_use_trt,
              FLAGS_gpu_id,
-             FLAGS_key,
-             FLAGS_batch_size);
+             FLAGS_key);
 
   double total_running_time_s = 0.0;
   double total_imread_time_s = 0.0;

+ 1 - 2
deploy/cpp/demo/segmenter.cpp

@@ -59,8 +59,7 @@ int main(int argc, char** argv) {
              FLAGS_use_gpu,
              FLAGS_use_trt,
              FLAGS_gpu_id,
-             FLAGS_key,
-             FLAGS_batch_size);
+             FLAGS_key);
 
   double total_running_time_s = 0.0;
   double total_imread_time_s = 0.0;