yjh0410 1 년 전
부모
커밋
73d8db08da
3개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  1. 1 1
      yolo/models/yolov3/yolov3_backbone.py
  2. 1 1
      yolo/models/yolov5/yolov5_backbone.py
  3. 1 1
      yolo/models/yolov5_af/yolov5_af_backbone.py

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

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

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

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

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

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