feat: shifting to yolo11

.gitconfig

@@ -0,0 +1,5 @@
+[http]
+        sslVerify = false
+        postBuffer = 1048576000
+[url "https://gitclone.com/"]
+        insteadOf = https://

.gitignore

@@ -24,3 +24,4 @@ config.ts
 .ipynb_checkpoints/
 env
 .vscode
+.gitconfig

data/config.demo.ts

@@ -1,16 +1,16 @@
 export const config = {
-    // laf云后台的appid
-    LAF_APPID: "your_laf_appid",
-    // 云函数getTempCOS调用得到
-    // 如果报错403，需要加laf环境变量，DEV_IPS="xxx.xxx.xxx.xxx"，其中xxx为调用本段代码的机器公网ip
-    COS_KEY: {
-        "ExpiredTime": 1687775333,
-        "Expiration": "2023-06-26T10:28:53Z",
-        "Credentials": {
-            "Token": "your_token",
-            "TmpSecretId": "your_secret_id",
-            "TmpSecretKey": "your_secret_key"
-        },
-        "RequestId": "0c56a822-7d46-4256-8446-8c038ac32ce6"
-    }
-}
+  // laf云后台的appid
+  LAF_APPID: "your_laf_appid",
+  // 云函数getTempCOS调用得到
+  // 如果报错403，需要加laf环境变量，DEV_IPS="xxx.xxx.xxx.xxx"，其中xxx为调用本段代码的机器公网ip
+  COS_KEY: {
+      "ExpiredTime": 1687775333,
+      "Expiration": "2023-06-26T10:28:53Z",
+      "Credentials": {
+          "Token": "your_token",
+          "TmpSecretId": "your_secret_id",
+          "TmpSecretKey": "your_secret_key"
+      },
+      "RequestId": "0c56a822-7d46-4256-8446-8c038ac32ce6"
+  }
+}

data_preprocess.py

@@ -3,13 +3,15 @@
 import torch
 from tqdm import tqdm
 from PIL import Image
+from ultralytics import YOLO
 
 SRC = "data/photos"
 DEST = "data/crop_photos"
 
 if __name__ == "__main__":
-    print("loading YOLOv5 model...")
-    model = torch.hub.load("yolov5", "custom", "yolov5/yolov5m.pt", source="local")
+    print("loading YOLO11 model...")
+    model = YOLO("yolo11m-cls.pt")
+    # model = torch.hub.load("yolov5", "custom", "yolov5/yolov5m.pt", source="local")
 
     num_photos = 0
     num_skipped_photos = 0

install_nodejs.sh

@@ -9,3 +9,4 @@ echo 'export PATH=$NODE_HOME/bin:$PATH' | tee -a /etc/profile
 source /etc/profile
 node -v
 npm -v
+npm config set registry https://registry.npmmirror.com

train.py

@@ -23,7 +23,7 @@ def main():
     )
     args = parser.parse_args()
 
-    model = YOLO("yolov8m-cls.pt")
+    model = YOLO("yolo11m-cls.pt")
     export_dir = "./export"
 
     results = model.train(data=f"{args.data}", epochs=args.epoch, imgsz=args.size)