Explorar el Código

add darknet53_silu ckpt for YOLOv3

yjh0410 hace 1 año
padre
commit
740c105d9c
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  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",
 }