wangjin2000 commited on
Commit
55ade3c
·
verified ·
1 Parent(s): efb0a6f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ model.eval() # Set the model to evaluation mode
52
  def preprocess_image(image):
53
  #img0 = cv2.imread(image_path)
54
  #img = letterbox(img0, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
55
- img0 = letterbox(imgge, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
56
  img = img.transpose(2, 0, 1)[::-1] # Convert BGR to RGB, to 3x416x416
57
  img = np.ascontiguousarray(img)
58
  img = torch.from_numpy(img).to(device)
 
52
  def preprocess_image(image):
53
  #img0 = cv2.imread(image_path)
54
  #img = letterbox(img0, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
55
+ img0 = letterbox(image, 640, stride=32, auto=True)[0] # Resize and pad to 640x640
56
  img = img.transpose(2, 0, 1)[::-1] # Convert BGR to RGB, to 3x416x416
57
  img = np.ascontiguousarray(img)
58
  img = torch.from_numpy(img).to(device)