Skip to content

GitLab

"...data/git@developer.sourcefind.cn:dcuai/dlexamples.git" did not exist on "a30b77fe6ad985bce0e362e4715a6a21a7ed488e"
Commit ff446b76 authored by LDOUBLEV's avatar LDOUBLEV
Browse files

int to int64 of lod type

parent 6420a76b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
deploy/lite/ocr_db_crnn.cc
View file @ ff446b76
...@@ -142,14 +142,13 @@ void RunRecModel(std::vector<std::vector<std::vector<int>>> boxes, cv::Mat img, ...@@ -142,14 +142,13 @@ void RunRecModel(std::vector<std::vector<std::vector<int>>> boxes, cv::Mat img,
// Get output and run postprocess // Get output and run postprocess
std::unique_ptr<const Tensor> output_tensor0( std::unique_ptr<const Tensor> output_tensor0(
std::move(predictor_crnn->GetOutput(0))); std::move(predictor_crnn->GetOutput(0)));
auto *rec_idx = output_tensor0->data<int>(); auto *rec_idx = output_tensor0->data<int64>();
auto rec_idx_lod = output_tensor0->lod(); auto rec_idx_lod = output_tensor0->lod();
auto shape_out = output_tensor0->shape(); auto shape_out = output_tensor0->shape();
std::vector<int> pred_idx; std::vector<int> pred_idx;
for (int n = int(rec_idx_lod[0][0]); n < int(rec_idx_lod[0][1] * 2); for (int n = int(rec_idx_lod[0][0]); n < int(rec_idx_lod[0][1]); n += 1) {
n += 2) {
pred_idx.push_back(int(rec_idx[n])); pred_idx.push_back(int(rec_idx[n]));
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment