提交 d1de7db0 编写于 作者: qnqinan's avatar qnqinan

update

上级 7aad1213
...@@ -27,19 +27,19 @@ static const char *g_densebox_combine = "../models/densebox"; ...@@ -27,19 +27,19 @@ static const char *g_densebox_combine = "../models/densebox";
int main() { int main() {
paddle_mobile::fpga::open_device(); paddle_mobile::fpga::open_device();
paddle_mobile::PaddleMobile<paddle_mobile::FPGA> paddle_mobile; paddle_mobile::PaddleMobile<paddle_mobile::FPGA> paddle_mobile;
//paddle_mobile.SetThreadNum(4); // paddle_mobile.SetThreadNum(4);
if (paddle_mobile.Load(std::string(g_densebox_combine) + "/model", if (paddle_mobile.Load(std::string(g_densebox_combine) + "/model",
std::string(g_densebox_combine) + "/params", true)) { std::string(g_densebox_combine) + "/params", true)) {
//std::vector<float> input; // std::vector<float> input;
//std::vector<int64_t> dims{1, 3, 512, 1024}; // std::vector<int64_t> dims{1, 3, 512, 1024};
//GetInput<float>(g_test_image_1x3x224x224_banana, &input, dims); // GetInput<float>(g_test_image_1x3x224x224_banana, &input, dims);
//auto vec_result = paddle_mobile.Predict(input, dims); // auto vec_result = paddle_mobile.Predict(input, dims);
Tensor input_tensor; Tensor input_tensor;
SetupTensor<float>(&input_tensor, {1, 3, 512, 1024}, static_cast<float>(0), SetupTensor<float>(&input_tensor, {1, 3, 512, 1024}, static_cast<float>(0),
static_cast<float>(1)); static_cast<float>(1));
//readStream(g_image_src_float, // readStream(g_image_src_float,
// input_tensor.mutable_data<float>({1, 3, 224, 224})); // input_tensor.mutable_data<float>({1, 3, 224, 224}));
paddle_mobile.FeedData(input_tensor); paddle_mobile.FeedData(input_tensor);
paddle_mobile.Predict_To(-1); paddle_mobile.Predict_To(-1);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册