wangjin2000 commited on
Commit
e25af60
·
verified ·
1 Parent(s): 64d21f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -51,7 +51,7 @@ model.eval() # Set the model to evaluation mode
51
  #def preprocess_image(image_path):
52
  def preprocess_image(image):
53
  #img0 = cv2.imread(image_path)
54
- print("in preprocess-0 image.shape:",image.shape)
55
  img = letterbox(image, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
56
  #img = letterbox(img0, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
57
  print("in preprocess-1 img.shape:",img.shape)
 
51
  #def preprocess_image(image_path):
52
  def preprocess_image(image):
53
  #img0 = cv2.imread(image_path)
54
+ print("in preprocess-0 image.shape:",image.size)
55
  img = letterbox(image, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
56
  #img = letterbox(img0, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
57
  print("in preprocess-1 img.shape:",img.shape)