未验证 提交 eb1fe1ca 编写于 作者: Z Zhang Jun 提交者: GitHub

Merge branch 'develop' into devel-fix

...@@ -18,7 +18,7 @@ import sys ...@@ -18,7 +18,7 @@ import sys
import numpy as np import numpy as np
preprocess = Sequential([ preprocess = Sequential([
File2Image(), BGR2RGB(), File2Image(), BGR2RGB(),
Normalize([123.675, 116.28, 103.53], [58.395, 57.12, 57.375], False), Normalize([123.675, 116.28, 103.53], [58.395, 57.12, 57.375], False),
Resize((512, 512)), Transpose((2, 0, 1)) Resize((512, 512)), Transpose((2, 0, 1))
]) ])
...@@ -33,6 +33,7 @@ im = preprocess(sys.argv[1]) ...@@ -33,6 +33,7 @@ im = preprocess(sys.argv[1])
fetch_map = client.predict( fetch_map = client.predict(
feed={ feed={
"image": im, "image": im,
"im_shape": np.array([512, 512]),
"scale_factor": np.array([1.0, 1.0]).reshape(-1), "scale_factor": np.array([1.0, 1.0]).reshape(-1),
}, },
fetch=["save_infer_model/scale_0.tmp_1"], fetch=["save_infer_model/scale_0.tmp_1"],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册