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

imp prior_box kernel and add superrelution test file

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