Browse Source

add YOLOv8 backbone in1k pretrained weight

yjh0410 1 year ago
parent
commit
a8ca483ef5
1 changed files with 2 additions and 2 deletions
  1. 2 2
      yolo/models/yolov8/yolov8_backbone.py

+ 2 - 2
yolo/models/yolov8/yolov8_backbone.py

@@ -10,8 +10,8 @@ except:
 pretrained_urls = {
     'n': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elandarknet_n_in1k_62.1.pth",
     's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elandarknet_s_in1k_71.3.pth",
-    'm': None,
-    'l': None,
+    'm': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elandarknet_m_in1k_75.7.pth",
+    'l': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/elandarknet_l_in1k_77.3.pth",
     'x': None,
 }