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

fix use_cv2 of SSD in infer (#1264)

上级 b53d290d
......@@ -134,7 +134,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 300
use_cv2: false
use_cv2: true
- !Permute {}
- !NormalizeImage
is_scale: false
......
......@@ -140,7 +140,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 300
use_cv2: false
use_cv2: true
- !Permute
to_bgr: false
- !NormalizeImage
......
......@@ -140,7 +140,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 300
use_cv2: false
use_cv2: true
- !Permute
to_bgr: false
- !NormalizeImage
......
......@@ -142,7 +142,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 512
use_cv2: false
use_cv2: true
- !Permute
to_bgr: false
- !NormalizeImage
......
......@@ -144,7 +144,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 512
use_cv2: false
use_cv2: true
- !Permute
to_bgr: false
- !NormalizeImage
......
......@@ -150,7 +150,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 320
use_cv2: false
use_cv2: true
- !NormalizeImage
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
......
......@@ -147,7 +147,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 300
use_cv2: false
use_cv2: true
- !NormalizeImage
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
......
......@@ -151,7 +151,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 320
use_cv2: false
use_cv2: true
- !NormalizeImage
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
......
......@@ -158,7 +158,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 320
use_cv2: false
use_cv2: true
- !NormalizeImage
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
......
......@@ -151,7 +151,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 320
use_cv2: false
use_cv2: true
- !NormalizeImage
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
......
......@@ -158,7 +158,7 @@ TestReader:
interp: 1
max_size: 0
target_size: 320
use_cv2: false
use_cv2: true
- !NormalizeImage
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
......
......@@ -25,6 +25,7 @@ DEFINE_string(model_dir, "", "Path of inference model");
DEFINE_string(image_path, "", "Path of input image");
DEFINE_string(video_path, "", "Path of input video");
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_int32(gpu_id, 0, "Device id of GPU to execute");
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.
先完成此消息的编辑!
想要评论请 注册