yjh0410 1 year ago
parent
commit
73d8db08da

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

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

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

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

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

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