diff --git a/python/examples/detection/ssd_vgg16_300_240e_voc/test_client.py b/python/examples/detection/ssd_vgg16_300_240e_voc/test_client.py index 59024d010a27c1569b5a07afd4508ad19894d89e..490ee590d01746102d8a04f26a15a65ab828b393 100755 --- a/python/examples/detection/ssd_vgg16_300_240e_voc/test_client.py +++ b/python/examples/detection/ssd_vgg16_300_240e_voc/test_client.py @@ -18,7 +18,7 @@ import sys import numpy as np preprocess = Sequential([ - File2Image(), BGR2RGB(), + File2Image(), BGR2RGB(), Normalize([123.675, 116.28, 103.53], [58.395, 57.12, 57.375], False), Resize((512, 512)), Transpose((2, 0, 1)) ]) @@ -33,8 +33,9 @@ im = preprocess(sys.argv[1]) fetch_map = client.predict( feed={ "image": im, + "im_shape": np.array([512, 512]), "scale_factor": np.array([1.0, 1.0]).reshape(-1), }, fetch=["save_infer_model/scale_0.tmp_1"], batch=False) -print(fetch_map) +print('ysl-----', fetch_map)