add yologpu test file

test/CMakeLists.txt

@@ -257,6 +257,10 @@ else ()
     ADD_EXECUTABLE(test-mobilenetgpu  net/test_mobilenet_GPU.cpp test_helper.h  test_include.h)
     target_link_libraries(test-mobilenetgpu paddle-mobile)
 
+    # gen test
+    ADD_EXECUTABLE(test-yologpu net/test_yologpu.cpp test_helper.h  test_include.h executor_for_test.h)
+    target_link_libraries(test-yologpu paddle-mobile)
+
 
     #add_library(test-lib-size SHARED common/test_lib_size.h common/test_lib_size.cpp)
 

test/net/test_yologpu.cpp

+/* Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License. */
+
+#include <iostream>
+#include "../test_helper.h"
+#include "../test_include.h"
+
+int main() {
+    paddle_mobile::PaddleMobile<paddle_mobile::GPU_CL> paddle_mobile;
+    //    paddle_mobile.SetThreadNum(4);
+    auto time1 = paddle_mobile::time();
+    //  auto isok = paddle_mobile.Load(std::string(g_mobilenet_detect) + "/model",
+    //                     std::string(g_mobilenet_detect) + "/params", true);
+
+    auto isok = paddle_mobile.Load(g_yolo_mul, true);
+    if (isok) {
+        auto time2 = paddle_mobile::time();
+        std::cout << "load cost :" << paddle_mobile::time_diff(time1, time2) << "ms"
+                  << std::endl;
+
+        std::vector<float> input;
+        std::vector<int64_t> dims{1, 3, 416, 416};
+        GetInput<float>(g_yolo_img, &input, dims);
+
+        std::vector<float> vec_result;
+        //            = paddle_mobile.Predict(input, dims);
+
+        auto time3 = paddle_mobile::time();
+        int max = 10;
+        for (int i = 0; i < max; ++i) {
+            vec_result = paddle_mobile.Predict(input, dims);
+        }
+        auto time4 = paddle_mobile::time();
+
+        //    auto time3 = paddle_mobile::time();
+
+        //    for (int i = 0; i < 10; ++i) {
+        //      auto vec_result = paddle_mobile.Predict(input, dims);
+        //    }
+
+        //    auto time4 = paddle_mobile::time();
+
+        std::cout << "predict cost :"
+                  << paddle_mobile::time_diff(time3, time4) / max << "ms"
+                  << std::endl;
+        std::vector<float>::iterator biggest =
+                std::max_element(std::begin(vec_result), std::end(vec_result));
+        std::cout << " Max element is " << *biggest << " at position "
+                  << std::distance(std::begin(vec_result), biggest) << std::endl;
+//        for (float i : vec_result) {
+//            std::cout << i << std::endl;
+//        }
+    }
+    return 0;
+}

test/test_helper.h

@@ -34,6 +34,7 @@ static const char *g_mobilenet_detect = "../models/mobilenet-detect";
 static const char *g_squeezenet = "../models/squeezenet";
 static const char *g_googlenet = "../models/googlenet";
 static const char *g_mobilenet = "../models/mobilenet";
+static const char *g_mobilenet_mul = "../models/mobilenet_mul";
 static const char *g_alexnet = "../models/alexnet";
 static const char *g_inceptionv4 = "../models/inceptionv4";
 static const char *g_nlp = "../models/nlp";
@@ -41,6 +42,7 @@ static const char *g_resnet_50 = "../models/resnet_50";
 static const char *g_resnet = "../models/resnet";
 static const char *g_googlenet_combine = "../models/googlenet_combine";
 static const char *g_yolo = "../models/yolo";
+static const char *g_yolo_mul = "../models/yolo_mul";
 static const char *g_fluid_fssd_new = "../models/fluid_fssd_new";
 
 static const char *g_test_image_1x3x224x224 =
@@ -51,6 +53,8 @@ static const char *g_hand = "../images/hand_image";
 static const char *g_imgfssd_ar = "../images/test_image_ssd_ar";
 static const char *g_imgfssd_ar1 = "../images/003_0001.txt";
 static const char *g_img = "../images/img.bin";
+static const char *g_yolo_img = "../images/in_put_1_3_416_416_2";
+static const char *g_mobilenet_img = "../images/image";
 
 using paddle_mobile::framework::DDim;
 using paddle_mobile::framework::Tensor;