yjh0410 1 yıl önce
ebeveyn
işleme
23140f638a

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

@@ -11,7 +11,7 @@ pretrained_urls = {
     '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,
+    'l': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/elandarknet_l_in1k_77.3.pth",
     'x': None,
 }
 

+ 1 - 1
yolo/models/yolov8_e2e/build.py

@@ -1,7 +1,7 @@
 import torch.nn as nn
 
 from .loss import SetCriterion
-from .yolov8 import Yolov8E2E
+from .yolov8_e2e import Yolov8E2E
 
 
 # build object detector

+ 1 - 1
yolo/models/yolov8_e2e/yolov8_backbone.py

@@ -11,7 +11,7 @@ pretrained_urls = {
     '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,
+    'l': "https://github.com/yjh0410/YOLO-Tutorial-v2/releases/download/in1k_pretrained_weight/elandarknet_l_in1k_77.3.pth",
     'x': None,
 }
 

+ 0 - 0
yolo/models/yolov8_e2e/yolov8.py → yolo/models/yolov8_e2e/yolov8_e2e.py