未验证 提交 a30abac2 编写于 作者: G Guanghua Yu 提交者: GitHub

fix use_cv2 of SSD in infer (#1264)

上级 b53d290d
...@@ -134,7 +134,7 @@ TestReader: ...@@ -134,7 +134,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 300 target_size: 300
use_cv2: false use_cv2: true
- !Permute {} - !Permute {}
- !NormalizeImage - !NormalizeImage
is_scale: false is_scale: false
......
...@@ -140,7 +140,7 @@ TestReader: ...@@ -140,7 +140,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 300 target_size: 300
use_cv2: false use_cv2: true
- !Permute - !Permute
to_bgr: false to_bgr: false
- !NormalizeImage - !NormalizeImage
......
...@@ -140,7 +140,7 @@ TestReader: ...@@ -140,7 +140,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 300 target_size: 300
use_cv2: false use_cv2: true
- !Permute - !Permute
to_bgr: false to_bgr: false
- !NormalizeImage - !NormalizeImage
......
...@@ -142,7 +142,7 @@ TestReader: ...@@ -142,7 +142,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 512 target_size: 512
use_cv2: false use_cv2: true
- !Permute - !Permute
to_bgr: false to_bgr: false
- !NormalizeImage - !NormalizeImage
......
...@@ -144,7 +144,7 @@ TestReader: ...@@ -144,7 +144,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 512 target_size: 512
use_cv2: false use_cv2: true
- !Permute - !Permute
to_bgr: false to_bgr: false
- !NormalizeImage - !NormalizeImage
......
...@@ -150,7 +150,7 @@ TestReader: ...@@ -150,7 +150,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 320 target_size: 320
use_cv2: false use_cv2: true
- !NormalizeImage - !NormalizeImage
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
......
...@@ -147,7 +147,7 @@ TestReader: ...@@ -147,7 +147,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 300 target_size: 300
use_cv2: false use_cv2: true
- !NormalizeImage - !NormalizeImage
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
......
...@@ -151,7 +151,7 @@ TestReader: ...@@ -151,7 +151,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 320 target_size: 320
use_cv2: false use_cv2: true
- !NormalizeImage - !NormalizeImage
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
......
...@@ -158,7 +158,7 @@ TestReader: ...@@ -158,7 +158,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 320 target_size: 320
use_cv2: false use_cv2: true
- !NormalizeImage - !NormalizeImage
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
......
...@@ -151,7 +151,7 @@ TestReader: ...@@ -151,7 +151,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 320 target_size: 320
use_cv2: false use_cv2: true
- !NormalizeImage - !NormalizeImage
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
......
...@@ -158,7 +158,7 @@ TestReader: ...@@ -158,7 +158,7 @@ TestReader:
interp: 1 interp: 1
max_size: 0 max_size: 0
target_size: 320 target_size: 320
use_cv2: false use_cv2: true
- !NormalizeImage - !NormalizeImage
mean: [0.485, 0.456, 0.406] mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225] std: [0.229, 0.224, 0.225]
......
...@@ -25,6 +25,7 @@ DEFINE_string(model_dir, "", "Path of inference model"); ...@@ -25,6 +25,7 @@ DEFINE_string(model_dir, "", "Path of inference model");
DEFINE_string(image_path, "", "Path of input image"); DEFINE_string(image_path, "", "Path of input image");
DEFINE_string(video_path, "", "Path of input video"); DEFINE_string(video_path, "", "Path of input video");
DEFINE_bool(use_gpu, false, "Infering with GPU or CPU"); DEFINE_bool(use_gpu, false, "Infering with GPU or CPU");
DEFINE_bool(use_camera, false, "Use camera or not");
DEFINE_string(run_mode, "fluid", "Mode of running(fluid/trt_fp32/trt_fp16)"); DEFINE_string(run_mode, "fluid", "Mode of running(fluid/trt_fp32/trt_fp16)");
DEFINE_int32(gpu_id, 0, "Device id of GPU to execute"); DEFINE_int32(gpu_id, 0, "Device id of GPU to execute");
DEFINE_int32(camera_id, -1, "Device id of camera to predict"); DEFINE_int32(camera_id, -1, "Device id of camera to predict");
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册