yjh0410 1 year ago
parent
commit
a34c3979e0

+ 2 - 2
yolo/models/gelan/gelan_backbone.py

@@ -8,8 +8,8 @@ except:
 
 # IN1K pretrained weight
 pretrained_urls = {
-    's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/gelan_s_in1k_68.4.pth",
-    'c': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/gelan_c_in1k_76.7.pth",
+    's': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/gelan_s_in1k_68.4.pth",
+    'c': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/gelan_c_in1k_76.7.pth",
 }
 
 # ----------------- GELAN backbone proposed by YOLOv9 -----------------

+ 1 - 1
yolo/models/yolov3/yolov3_backbone.py

@@ -9,7 +9,7 @@ except:
 # IN1K pretrained weight
 pretrained_urls = {
     'n': None,
-    's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/darknet_s_in1k_68.5.pth",
+    's': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/darknet_s_in1k_68.5.pth",
     'm': None,
     'l': None,
     'x': None,

+ 2 - 2
yolo/models/yolov5/yolov5_backbone.py

@@ -9,8 +9,8 @@ except:
 # IN1K pretrained weight
 pretrained_urls = {
     'n': None,
-    's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/cspdarknet_s_in1k_70.1.pth",
-    'm': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/cspdarknet_m_in1k_75.2.pth",
+    's': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/cspdarknet_s_in1k_70.1.pth",
+    'm': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/cspdarknet_m_in1k_75.2.pth",
     'l': None,
     'x': None,
 }

+ 2 - 2
yolo/models/yolov5_af/yolov5_af_backbone.py

@@ -9,8 +9,8 @@ except:
 # IN1K pretrained weight
 pretrained_urls = {
     'n': None,
-    's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/cspdarknet_s_in1k_70.1.pth",
-    'm': None,
+    's': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/cspdarknet_s_in1k_70.1.pth",
+    'm': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/cspdarknet_m_in1k_75.2.pth",
     'l': None,
     'x': None,
 }

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

@@ -8,10 +8,10 @@ except:
 
 # IN1K pretrained weight
 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': "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",
+    'n': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/elandarknet_n_in1k_62.1.pth",
+    's': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/elandarknet_s_in1k_71.3.pth",
+    'm': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/elandarknet_m_in1k_75.7.pth",
+    'l': None,
     'x': None,
 }