Home Explore Help
Sign In
junxiaoyao
/
YOLO-Tutorial-v2
1
0
Fork 0
Files Issues 0 Pull Requests 0 Wiki
Browse Source

train v3 & v5 with IN1k pretrained

yjh0410 1 year ago
parent
91e8717542
commit
00d2bfb674
2 changed files with 2 additions and 2 deletions
Split View Show Diff Stats
  1. 1 1
      yolo/models/yolov3/yolov3_backbone.py
  2. 1 1
      yolo/models/yolov5/yolov5_backbone.py

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

@@ -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_44.2.pth",
 
+    's': "https://github.com/yjh0410/ICLab/releases/download/in1k_pretrained/darknet_s_in1k_68.5.pth",
 
     'm': None,
 
     'l': None,
 
     'x': None,

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

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