浏览代码

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",
 }