Skip to content

GitLab

Commit 1751be75 authored by gaotingquan's avatar gaotingquan
Browse files

Fix the bug about raising error when reading image failed

parent add9faa8
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
deploy/cpp_infer/src/main.cpp
View file @ 1751be75
...@@ -50,6 +50,11 @@ int main(int argc, char **argv) { ...@@ -50,6 +50,11 @@ int main(int argc, char **argv) {
cv::Mat srcimg = cv::imread(img_path, cv::IMREAD_COLOR); cv::Mat srcimg = cv::imread(img_path, cv::IMREAD_COLOR);
if (!srcimg.data) {
std::cerr << "[ERROR] image read failed! image path: " << img_path << "\n";
exit(1);
}
DBDetector det(config.det_model_dir, config.use_gpu, config.gpu_id, DBDetector det(config.det_model_dir, config.use_gpu, config.gpu_id,
config.gpu_mem, config.cpu_math_library_num_threads, config.gpu_mem, config.cpu_math_library_num_threads,
config.use_mkldnn, config.max_side_len, config.det_db_thresh, config.use_mkldnn, config.max_side_len, config.det_db_thresh,
......
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