Переглянути джерело

add darknet53_silu ckpt for YOLOv3

yjh0410 1 рік тому
батько
коміт
740c105d9c
1 змінених файлів з 1 додано та 1 видалено
  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",
 }