提交 08b1b21a 编写于 作者: Y yangfei

imp prior_box kernel and add superrelution test file

上级 1acaf4f6
...@@ -39,8 +39,8 @@ void PriorBoxKernel<GPU_CL, float>::Compute( ...@@ -39,8 +39,8 @@ void PriorBoxKernel<GPU_CL, float>::Compute(
const auto &input_aspect_ratio = param.AspectRatios(); const auto &input_aspect_ratio = param.AspectRatios();
const bool &flip = param.Flip(); const bool &flip = param.Flip();
const bool &clip = param.Clip(); const bool &clip = param.Clip();
int isclip =0; int isclip = 0;
if(clip){ if (clip) {
isclip = 1; isclip = 1;
} }
const float &step_w = param.StepW(); const float &step_w = param.StepW();
...@@ -146,11 +146,11 @@ void PriorBoxKernel<GPU_CL, float>::Compute( ...@@ -146,11 +146,11 @@ void PriorBoxKernel<GPU_CL, float>::Compute(
DLOG << "num_priors:" << num_priors; DLOG << "num_priors:" << num_priors;
DLOG << "C:" << C; DLOG << "C:" << C;
DLOG << "isclip:" << isclip; DLOG << "isclip:" << isclip;
for(int i=0;i<num_priors;i++){ for (int i = 0; i < num_priors; i++) {
DLOG<<box_width[i]; DLOG << box_width[i];
} }
for(int i=0;i<num_priors;i++){ for (int i = 0; i < num_priors; i++) {
DLOG<<box_height[i]; DLOG << box_height[i];
} }
status = clSetKernelArg(kernel, 0, sizeof(int), &c_block); status = clSetKernelArg(kernel, 0, sizeof(int), &c_block);
CL_CHECK_ERRORS(status); CL_CHECK_ERRORS(status);
......
...@@ -25,9 +25,9 @@ int main() { ...@@ -25,9 +25,9 @@ int main() {
paddle_mobile.SetCLPath("/data/local/tmp/bin"); paddle_mobile.SetCLPath("/data/local/tmp/bin");
#endif #endif
auto isok = auto isok = paddle_mobile.Load(std::string(g_super) + "/model",
paddle_mobile.Load(std::string(g_super) + "/model", std::string(g_super) + "/params", true, false,
std::string(g_super) + "/params",true, false,1,true); 1, true);
// auto isok = paddle_mobile.Load(std::string(g_mobilenet_mul), true); // auto isok = paddle_mobile.Load(std::string(g_mobilenet_mul), true);
if (isok) { if (isok) {
...@@ -36,16 +36,15 @@ int main() { ...@@ -36,16 +36,15 @@ int main() {
<< std::endl; << std::endl;
std::vector<float> input; std::vector<float> input;
std::vector<int64_t> dims{1, 1,300, 300}; std::vector<int64_t> dims{1, 1, 300, 300};
GetInput<float>(g_yolo_img, &input, dims); GetInput<float>(g_yolo_img, &input, dims);
std::vector<float> vec_result ; std::vector<float> vec_result;
auto time3 = paddle_mobile::time(); auto time3 = paddle_mobile::time();
int max = 10; int max = 10;
for (int i = 0; i < max; ++i) { for (int i = 0; i < max; ++i) {
vec_result = paddle_mobile.Predict(input, dims); vec_result = paddle_mobile.Predict(input, dims);
} }
auto time4 = paddle_mobile::time(); auto time4 = paddle_mobile::time();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册