test_super.cpp 2.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20
/* 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 "../../src/common/types.h"
#include "../test_helper.h"
#include "../test_include.h"

int main() {
L
liuruilong 已提交
21 22 23
  paddle_mobile::PaddleMobileConfigInternal config;
  config.load_when_predict = true;

xiebaiyuan's avatar
xiebaiyuan 已提交
24
  paddle_mobile::PaddleMobile<paddle_mobile::GPU_CL> paddle_mobile(config);
25 26
  //    paddle_mobile.SetThreadNum(4);
  auto time1 = paddle_mobile::time();
27
#ifdef PADDLE_MOBILE_CL
28
  paddle_mobile.SetCLPath("/data/local/tmp/bin");
29 30
#endif

31 32
  auto isok = paddle_mobile.Load(std::string(g_super) + "/model",
                                 std::string(g_super) + "/params", true, false,
L
liuruilong 已提交
33
                                 1, false);
34

35 36 37 38 39
  //  auto isok = paddle_mobile.Load(std::string(g_mobilenet_mul), true);
  if (isok) {
    auto time2 = paddle_mobile::time();
    std::cout << "load cost :" << paddle_mobile::time_diff(time1, time2) << "ms"
              << std::endl;
40

41 42
    std::vector<float> input;
    std::vector<int64_t> dims{1, 1, 300, 300};
xiebaiyuan's avatar
xiebaiyuan 已提交
43
    GetInput<float>(g_test_image_1x3x224x224, &input, dims);
44

45
    std::vector<float> vec_result;
46

47
    auto time3 = paddle_mobile::time();
xiebaiyuan's avatar
xiebaiyuan 已提交
48 49
    int max = 10;

50
    for (int i = 0; i < max; ++i) {
xiebaiyuan's avatar
xiebaiyuan 已提交
51
      auto time5 = paddle_mobile::time();
52
      vec_result = paddle_mobile.Predict(input, dims);
xiebaiyuan's avatar
xiebaiyuan 已提交
53 54 55
      auto time6 = paddle_mobile::time();
      std::cout << "predict cost :第" << i << ": "
                << paddle_mobile::time_diff(time5, time6) << "ms" << std::endl;
56
    }
57
    auto time4 = paddle_mobile::time();
58

59 60
    std::cout << "predict cost :"
              << paddle_mobile::time_diff(time3, time4) / max << "ms"
61
              << std::endl;
xiebaiyuan's avatar
xiebaiyuan 已提交
62
    auto biggest =
63 64 65 66 67 68 69 70 71
        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;
  }

  std::cout << "如果结果Nan请查看: test/images/g_test_image_1x3x224x224_banana "
               "是否存在?"
            << std::endl;
  return 0;
72
}