tmpcode

model.py

@@ -6,9 +6,9 @@ import time
 
 
 def load_model():
 def load_model():
     # 定义预处理步骤
     # 定义预处理步骤
+    # 定义图像预处理步骤
     preprocess = transforms.Compose([
     preprocess = transforms.Compose([
-        transforms.Resize(256),
-        transforms.CenterCrop(224),
+        transforms.Resize((640, 480)),  # 调整图片大小
         transforms.ToTensor(),
         transforms.ToTensor(),
         transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
         transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
     ])
     ])