소스 검색

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