model loading update

.vscode/settings.json

@@ -0,0 +1,3 @@
+{
+    "python.formatting.provider": "black"
+}

app.py

@@ -78,13 +78,16 @@ def parse_args(known=False):
 def model_loading(model_name, device):
 
     # 加载本地模型
-    model = torch.hub.load(
-        local_model_path,
-        "custom",
-        path=f"{local_model_path}/{model_name}",
-        source="local",
-        device=device,
-    )
+    # model = torch.hub.load('ultralytics/yolov5', 'yolov5s', device=device)
+    model = torch.hub.load("ultralytics/yolov5", model_name, device=device)
+
+    # model = torch.hub.load(
+    #     local_model_path,
+    #     "custom",
+    #     path=f"{local_model_path}/{model_name}",
+    #     source="local",
+    #     device=device,
+    # )
 
     return model
 

requirements.txt

@@ -0,0 +1,35 @@
+# Base ----------------------------------------
+matplotlib>=3.2.2
+numpy>=1.18.5
+opencv-python>=4.1.2
+Pillow>=7.1.2
+PyYAML>=5.3.1
+requests>=2.23.0
+scipy>=1.4.1
+torch>=1.7.0
+torchvision>=0.8.1
+tqdm>=4.41.0
+
+# Logging -------------------------------------
+tensorboard>=2.4.1
+# wandb
+
+# Plotting ------------------------------------
+pandas>=1.1.4
+seaborn>=0.11.0
+
+# Export --------------------------------------
+# coremltools>=4.1  # CoreML export
+# onnx>=1.9.0  # ONNX export
+# onnx-simplifier>=0.3.6  # ONNX simplifier
+# scikit-learn==0.19.2  # CoreML quantization
+# tensorflow>=2.4.1  # TFLite export
+# tensorflowjs>=3.9.0  # TF.js export
+# openvino-dev  # OpenVINO export
+
+# Extras --------------------------------------
+# albumentations>=1.0.3
+# Cython  # for pycocotools https://github.com/cocodataset/cocoapi/issues/172
+# pycocotools>=2.0  # COCO mAP
+# roboflow
+thop  # FLOPs computation