fix(yolo): 修复模型下载功能 - 只包含可用的YOLO11n系列模型

Author: Jarvis

available_models.json

@@ -0,0 +1,27 @@
+{
+  "yolo11n-cls": {
+    "url": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.onnx",
+    "size_mb": 10.769930839538574,
+    "release": "v8.3.0"
+  },
+  "yolo11n-obb": {
+    "url": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-obb.onnx",
+    "size_mb": 10.493975639343262,
+    "release": "v8.3.0"
+  },
+  "yolo11n-pose": {
+    "url": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.onnx",
+    "size_mb": 11.30433177947998,
+    "release": "v8.3.0"
+  },
+  "yolo11n-seg": {
+    "url": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.onnx",
+    "size_mb": 11.218754768371582,
+    "release": "v8.3.0"
+  },
+  "yolo11n": {
+    "url": "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n.onnx",
+    "size_mb": 10.423833847045898,
+    "release": "v8.3.0"
+  }
+}

available_models.rs

@@ -0,0 +1,47 @@
+
+// Rust代码生成 - 可用的模型URL映射
+// 自动从GitHub API获取
+
+#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
+pub struct ModelInfo {
+    pub url: String,
+    pub size_mb: f64,
+    pub release: String,
+}
+
+/// 获取所有可用的预训练模型列表
+pub fn get_available_models() -> std::collections::HashMap<String, ModelInfo> {
+    let mut models = std::collections::HashMap::new();
+    
+    models.insert("yolo11n".to_string(), ModelInfo {
+        url: "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n.onnx".to_string(),
+        size_mb: 10.423833847045898,
+        release: "v8.3.0".to_string(),
+    });
+    
+    models.insert("yolo11n-cls".to_string(), ModelInfo {
+        url: "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-cls.onnx".to_string(),
+        size_mb: 10.769930839538574,
+        release: "v8.3.0".to_string(),
+    });
+    
+    models.insert("yolo11n-obb".to_string(), ModelInfo {
+        url: "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-obb.onnx".to_string(),
+        size_mb: 10.493975639343262,
+        release: "v8.3.0".to_string(),
+    });
+    
+    models.insert("yolo11n-pose".to_string(), ModelInfo {
+        url: "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-pose.onnx".to_string(),
+        size_mb: 11.30433177947998,
+        release: "v8.3.0".to_string(),
+    });
+    
+    models.insert("yolo11n-seg".to_string(), ModelInfo {
+        url: "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n-seg.onnx".to_string(),
+        size_mb: 11.218754768371582,
+        release: "v8.3.0".to_string(),
+    });
+    
+    models
+}

available_models.txt

@@ -0,0 +1,4 @@
+# 可用的YOLO模型URL
+# 格式: 模型名称, URL, 大小(MB)
+
+yolo11n|https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo11n.onnx|10.42

check_all_yolo_models.py

@@ -0,0 +1,120 @@
+#!/usr/bin/env python3
+"""
+检查所有可用的YOLO模型URL
+"""
+import requests
+import time
+import concurrent.futures
+
+# YOLO11 URLs (最新版本)
+YOLO11_URLS = {
+    "yolo11n": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo11n.onnx",
+    "yolo11s": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo11s.onnx",
+    "yolo11m": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo11m.onnx",
+    "yolo11l": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo11l.onnx",
+    "yolo11x": "https://github.com/ultralytics/assets/releases/download/v8.4.0/yolo11x.onnx",
+}
+
+# YOLOv8 URLs (旧版本)
+YOLOV8_URLS = {
+    "yolov8n": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.onnx",
+    "yolov8s": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s.onnx",
+    "yolov8m": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8m.onnx",
+    "yolov8l": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l.onnx",
+    "yolov8x": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8x.onnx",
+}
+
+# YOLOv8 segmentation URLs
+YOLOV8_SEG_URLS = {
+    "yolov8n-seg": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n-seg.onnx",
+    "yolov8s-seg": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s-seg.onnx",
+    "yolov8m-seg": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8m-seg.onnx",
+    "yolov8l-seg": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-seg.onnx",
+    "yolov8x-seg": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8x-seg.onnx",
+}
+
+# YOLOv8 pose URLs
+YOLOV8_POSE_URLS = {
+    "yolov8n-pose": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n-pose.onnx",
+    "yolov8s-pose": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s-pose.onnx",
+    "yolov8m-pose": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8m-pose.onnx",
+    "yolov8l-pose": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-pose.onnx",
+    "yolov8x-pose": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8x-pose.onnx",
+}
+
+# YOLOv8 classify URLs
+YOLOV8_CLS_URLS = {
+    "yolov8n-cls": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n-cls.onnx",
+    "yolov8s-cls": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s-cls.onnx",
+    "yolov8m-cls": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8m-cls.onnx",
+    "yolov8l-cls": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l-cls.onnx",
+    "yolov8x-cls": "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8x-cls.onnx",
+}
+
+# 合并所有URL
+ALL_URLS = {}
+ALL_URLS.update(YOLO11_URLS)
+ALL_URLS.update(YOLOV8_URLS)
+ALL_URLS.update(YOLOV8_SEG_URLS)
+ALL_URLS.update(YOLOV8_POSE_URLS)
+ALL_URLS.update(YOLOV8_CLS_URLS)
+
+def check_url(name_url):
+    """检查单个URL是否有效"""
+    name, url = name_url
+    try:
+        response = requests.head(url, timeout=10, allow_redirects=True)
+        size_mb = float(response.headers.get('content-length', 0)) / (1024 * 1024)
+        status = "✓ OK" if response.status_code == 200 else f"✗ {response.status_code}"
+        print(f"{status:12} | {name:20} | {size_mb:8.2f} MB | {url}")
+        return (name, url, response.status_code, size_mb)
+    except Exception as e:
+        print(f"{'✗ ERROR':12} | {name:20} | {'':8} | {url} - {str(e)}")
+        return (name, url, 0, 0)
+
+def main():
+    print("=" * 120)
+    print("检查所有YOLO模型URL有效性")
+    print("=" * 120)
+    print(f"{'状态':12} | {'模型名称':20} | {'大小':8} | URL")
+    print("-" * 120)
+    
+    working = []
+    not_working = []
+    
+    # 使用线程池并发检查
+    with concurrent.futures.ThreadPoolExecutor(max_workers=5) as executor:
+        results = list(executor.map(check_url, ALL_URLS.items()))
+    
+    print("\n" + "=" * 120)
+    print("总结:")
+    for name, url, status, size in results:
+        if status == 200:
+            working.append((name, url, size))
+        else:
+            not_working.append((name, url, status))
+    
+    print(f"可用模型: {len(working)}/{len(ALL_URLS)}")
+    if working:
+        print("\n可用的模型:")
+        for name, url, size in working:
+            print(f"  - {name}: {size:.2f} MB")
+    
+    if not_working:
+        print(f"\n不可用的模型: {len(not_working)}")
+        for name, url, status in not_working:
+            print(f"  - {name}: HTTP {status}")
+    
+    print("=" * 120)
+    
+    # 保存可用的URL到文件
+    if working:
+        with open("available_models.txt", "w", encoding="utf-8") as f:
+            f.write("# 可用的YOLO模型URL\n")
+            f.write("# 格式: 模型名称, URL, 大小(MB)\n\n")
+            for name, url, size in working:
+                f.write(f"{name}|{url}|{size:.2f}\n")
+        print("\n可用的模型列表已保存到 available_models.txt")
+
+if __name__ == "__main__":
+    main()

check_multiple_sources.py

@@ -0,0 +1,128 @@
+#!/usr/bin/env python3
+"""
+检查多个YOLO模型下载源
+"""
+import requests
+import json
+
+# 定义要测试的模型
+models_to_test = [
+    # YOLO11 - GitHub (v8.3.0)
+    ("yolo11n", "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11n.onnx"),
+    ("yolo11s", "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11s.onnx"),
+    ("yolo11m", "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11m.onnx"),
+    ("yolo11l", "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11l.onnx"),
+    ("yolo11x", "https://github.com/ultralytics/assets/releases/download/v8.3.0/yolo11x.onnx"),
+    
+    # YOLOv8 - GitHub (v8.2.0)
+    ("yolov8n", "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8n.onnx"),
+    ("yolov8s", "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8s.onnx"),
+    ("yolov8m", "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8m.onnx"),
+    ("yolov8l", "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8l.onnx"),
+    ("yolov8x", "https://github.com/ultralytics/assets/releases/download/v8.2.0/yolov8x.onnx"),
+    
+    # ModelScope URLs (中国镜像)
+    ("yolo11n-modelscope", "https://www.modelscope.cn/models/AI-ModelScope/YOLOv11n/resolve/master/yolo11n.onnx"),
+    ("yolo11s-modelscope", "https://www.modelscope.cn/models/AI-ModelScope/YOLOv11s/resolve/master/yolo11s.onnx"),
+    ("yolov8n-modelscope", "https://www.modelscope.cn/models/AI-ModelScope/YOLOv8n/resolve/master/yolov8n.onnx"),
+    ("yolov8s-modelscope", "https://www.modelscope.cn/models/AI-ModelScope/YOLOv8s/resolve/master/yolov8s.onnx"),
+    ("yolov8m-modelscope", "https://www.modelscope.cn/models/AI-ModelScope/YOLOv8m/resolve/master/yolov8m.onnx"),
+]
+
+def check_url(name, url):
+    """检查URL是否可访问"""
+    try:
+        # 使用HEAD请求，只获取头部信息
+        response = requests.head(url, timeout=15, allow_redirects=True)
+        content_length = response.headers.get('content-length')
+        size_mb = float(content_length) / (1024 * 1024) if content_length else 0
+        
+        return {
+            'name': name,
+            'url': url,
+            'status': response.status_code,
+            'size_mb': size_mb,
+            'ok': response.status_code == 200
+        }
+    except Exception as e:
+        return {
+            'name': name,
+            'url': url,
+            'status': 0,
+            'size_mb': 0,
+            'ok': False,
+            'error': str(e)
+        }
+
+def main():
+    print("="*120)
+    print("YOLO模型多源检查")
+    print("="*120)
+    
+    results = []
+    for name, url in models_to_test:
+        print(f"检查 {name}...", end=" ", flush=True)
+        result = check_url(name, url)
+        results.append(result)
+        if result['ok']:
+            print(f"✓ {result['size_mb']:.2f} MB")
+        elif result['status'] == 0:
+            print(f"✗ 错误: {result.get('error', '未知')}")
+        else:
+            print(f"✗ HTTP {result['status']}")
+    
+    print("\n" + "="*120)
+    print("总结:")
+    print("="*120)
+    
+    working = [r for r in results if r['ok']]
+    not_working = [r for r in results if not r['ok']]
+    
+    print(f"\n可用的模型: {len(working)}/{len(results)}")
+    if working:
+        print("\n可用的下载源:")
+        for r in working:
+            print(f"  ✓ {r['name']:30} | {r['size_mb']:7.2f} MB | {r['url']}")
+    
+    if not_working:
+        print(f"\n不可用的下载源: {len(not_working)}")
+        for r in not_working:
+            error_info = f" - {r.get('error', '')}" if r.get('error') else ""
+            print(f"  ✗ {r['name']:30} | HTTP {r['status']}{error_info}")
+    
+    # 保存结果
+    with open('model_urls_check.json', 'w', encoding='utf-8') as f:
+        json.dump({
+            'working': working,
+            'not_working': not_working
+        }, f, indent=2, ensure_ascii=False)
+    
+    print(f"\n✓ 检查结果已保存到 model_urls_check.json")
+    
+    # 生成Rust可用的代码
+    if working:
+        print("\n" + "="*120)
+        print("Rust代码 - 可用的模型:")
+        print("="*120)
+        print('''
+#[derive(Debug, Clone)]
+pub struct ModelDownloadInfo {
+    pub name: String,
+    pub url: String,
+    pub size_mb: f64,
+}
+
+pub fn get_available_model_urls() -> Vec<ModelDownloadInfo> {
+    vec![
+''')
+        for r in working:
+            print(f'        ModelDownloadInfo {{')
+            print(f'            name: "{r["name"]}".to_string(),')
+            print(f'            url: "{r["url"]}".to_string(),')
+            print(f'            size_mb: {r["size_mb"]:.2f},')
+            print(f'        }},')
+        print('    ]')
+        print('}')
+
+if __name__ == "__main__":
+    main()