Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
c8aa6d03
Commit
c8aa6d03
authored
Aug 10, 2021
by
MissPenguin
Browse files
fix
parent
4a182a89
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
9 additions
and
22 deletions
+9
-22
deploy/cpp_infer/CMakeLists.txt
deploy/cpp_infer/CMakeLists.txt
+0
-2
deploy/cpp_infer/CMakeLists_det.txt
deploy/cpp_infer/CMakeLists_det.txt
+0
-2
deploy/cpp_infer/CMakeLists_rec.txt
deploy/cpp_infer/CMakeLists_rec.txt
+0
-2
deploy/cpp_infer/CMakeLists_system.txt
deploy/cpp_infer/CMakeLists_system.txt
+0
-2
deploy/cpp_infer/include/ocr_det.h
deploy/cpp_infer/include/ocr_det.h
+9
-14
No files found.
deploy/cpp_infer/CMakeLists.txt
View file @
c8aa6d03
...
@@ -38,10 +38,8 @@ endif()
...
@@ -38,10 +38,8 @@ endif()
if
(
WIN32
)
if
(
WIN32
)
include_directories
(
"
${
PADDLE_LIB
}
/paddle/fluid/inference"
)
include_directories
(
"
${
PADDLE_LIB
}
/paddle/include"
)
include_directories
(
"
${
PADDLE_LIB
}
/paddle/include"
)
link_directories
(
"
${
PADDLE_LIB
}
/paddle/lib"
)
link_directories
(
"
${
PADDLE_LIB
}
/paddle/lib"
)
link_directories
(
"
${
PADDLE_LIB
}
/paddle/fluid/inference"
)
find_package
(
OpenCV REQUIRED PATHS
${
OPENCV_DIR
}
/build/ NO_DEFAULT_PATH
)
find_package
(
OpenCV REQUIRED PATHS
${
OPENCV_DIR
}
/build/ NO_DEFAULT_PATH
)
else
()
else
()
...
...
deploy/cpp_infer/CMakeLists_det.txt
View file @
c8aa6d03
...
@@ -38,10 +38,8 @@ endif()
...
@@ -38,10 +38,8 @@ endif()
if (WIN32)
if (WIN32)
include_directories("${PADDLE_LIB}/paddle/fluid/inference")
include_directories("${PADDLE_LIB}/paddle/include")
include_directories("${PADDLE_LIB}/paddle/include")
link_directories("${PADDLE_LIB}/paddle/lib")
link_directories("${PADDLE_LIB}/paddle/lib")
link_directories("${PADDLE_LIB}/paddle/fluid/inference")
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
else ()
else ()
...
...
deploy/cpp_infer/CMakeLists_rec.txt
View file @
c8aa6d03
...
@@ -38,10 +38,8 @@ endif()
...
@@ -38,10 +38,8 @@ endif()
if (WIN32)
if (WIN32)
include_directories("${PADDLE_LIB}/paddle/fluid/inference")
include_directories("${PADDLE_LIB}/paddle/include")
include_directories("${PADDLE_LIB}/paddle/include")
link_directories("${PADDLE_LIB}/paddle/lib")
link_directories("${PADDLE_LIB}/paddle/lib")
link_directories("${PADDLE_LIB}/paddle/fluid/inference")
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
else ()
else ()
...
...
deploy/cpp_infer/CMakeLists_system.txt
View file @
c8aa6d03
...
@@ -38,10 +38,8 @@ endif()
...
@@ -38,10 +38,8 @@ endif()
if (WIN32)
if (WIN32)
include_directories("${PADDLE_LIB}/paddle/fluid/inference")
include_directories("${PADDLE_LIB}/paddle/include")
include_directories("${PADDLE_LIB}/paddle/include")
link_directories("${PADDLE_LIB}/paddle/lib")
link_directories("${PADDLE_LIB}/paddle/lib")
link_directories("${PADDLE_LIB}/paddle/fluid/inference")
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
find_package(OpenCV REQUIRED PATHS ${OPENCV_DIR}/build/ NO_DEFAULT_PATH)
else ()
else ()
...
...
deploy/cpp_infer/include/ocr_det.h
View file @
c8aa6d03
...
@@ -38,20 +38,15 @@ namespace PaddleOCR {
...
@@ -38,20 +38,15 @@ namespace PaddleOCR {
class
DBDetector
{
class
DBDetector
{
public:
public:
explicit
DBDetector
(
const
std
::
string
&
model_dir
,
explicit
DBDetector
(
const
std
::
string
&
model_dir
,
const
bool
&
use_gpu
,
const
bool
&
use_gpu
=
false
,
const
int
&
gpu_id
,
const
int
&
gpu_mem
,
const
int
&
gpu_id
=
0
,
const
int
&
cpu_math_library_num_threads
,
const
int
&
gpu_mem
=
4000
,
const
bool
&
use_mkldnn
,
const
int
&
max_side_len
,
const
int
&
cpu_math_library_num_threads
=
1
,
const
double
&
det_db_thresh
,
const
bool
&
use_mkldnn
=
false
,
const
double
&
det_db_box_thresh
,
const
int
&
max_side_len
=
960
,
const
double
&
det_db_unclip_ratio
,
const
double
&
det_db_thresh
=
0.3
,
const
bool
&
use_polygon_score
,
const
bool
&
visualize
,
const
double
&
det_db_box_thresh
=
0.5
,
const
bool
&
use_tensorrt
,
const
bool
&
use_fp16
)
{
const
double
&
det_db_unclip_ratio
=
2.0
,
const
bool
&
use_polygon_score
=
false
,
const
bool
&
visualize
=
false
,
const
bool
&
use_tensorrt
=
false
,
const
bool
&
use_fp16
=
false
)
{
this
->
use_gpu_
=
use_gpu
;
this
->
use_gpu_
=
use_gpu
;
this
->
gpu_id_
=
gpu_id
;
this
->
gpu_id_
=
gpu_id
;
this
->
gpu_mem_
=
gpu_mem
;
this
->
gpu_mem_
=
gpu_mem
;
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment