Update test_client.py

python/examples/detection/ttfnet_darknet53_1x_coco/test_client.py

@@ -11,16 +11,18 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-
-from paddle_serving_client import Client
-from paddle_serving_app.reader import *
 import sys
 import numpy as np
+from paddle_serving_client import Client
+from paddle_serving_app.reader import *
+import cv2
 
 preprocess = Sequential([
-    File2Image(), BGR2RGB(), 
+        File2Image(),
+        Resize(
+        (512, 512), False, interpolation=cv2.INTER_LINEAR),
         Normalize([123.675, 116.28, 103.53], [58.395, 57.12, 57.375], False),
-    Resize((512, 512)), Transpose((2, 0, 1))
+        Transpose((2,0,1))
 ])
 
 postprocess = RCNNPostprocess("label_list.txt", "output")
@@ -29,11 +31,14 @@ client = Client()
 client.load_client_config("serving_client/serving_client_conf.prototxt")
 client.connect(['127.0.0.1:9494'])
 
-im = preprocess(sys.argv[1])
+im, im_info = preprocess(sys.argv[1])
+
+
 fetch_map = client.predict(
     feed={
         "image": im,
-        "scale_factor": np.array([1.0, 1.0]).reshape(-1),
+        "im_shape": np.array(list(im.shape[1:])).reshape(-1),
+        "scale_factor": im_info['scale_factor'],
     },
     fetch=["save_infer_model/scale_0.tmp_1"],
     batch=False)