Pārlūkot izejas kodu

add darknet53_silu ckpt for YOLOv3

yjh0410 1 gadu atpakaļ
vecāks
revīzija
740c105d9c
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      models/detectors/yolov3/yolov3_backbone.py

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

@@ -9,7 +9,7 @@ except:
 
 model_urls = {
     "darknet_tiny": "https://github.com/yjh0410/image_classification_pytorch/releases/download/weight/darknet_tiny.pth",
-    "darknet53": None,
+    "darknet53": "https://github.com/yjh0410/image_classification_pytorch/releases/download/weight/darknet53_silu.pth",
 }