Update python/run_axmodel.py

python/run_axmodel.py

@@ -22,8 +22,6 @@ def main(args):
     session = axe.InferenceSession(args.model_path)
     output_names = [x.name for x in session.get_outputs()]
     input_name = session.get_inputs()[0].name
-    print(input_name)
-    print(output_names)
 
     ori_image = cv2.imread(args.input_path)
     h, w = ori_image.shape[:2]
@@ -36,8 +34,6 @@ def main(args):
 
     #image = (image /1.0).astype(np.float32)
     image = np.transpose(np.expand_dims(np.ascontiguousarray(image), axis=0), (0,3,1,2))
-    print(image.shape)
-
     
     # Use the model to generate super-resolved images
     sr = session.run(output_names, {input_name: image})