wangjin2000
commited on
Update app.py
Browse files
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.
|
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)
|