Explorar el Código

release CSPDarkNet-Tiny

yjh0410 hace 2 años
padre
commit
65117d0b6c
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      models/detectors/yolov4/yolov4_backbone.py

+ 1 - 1
models/detectors/yolov4/yolov4_backbone.py

@@ -8,7 +8,7 @@ except:
     
 
 model_urls = {
-    "cspdarknet_tiny": None,
+    "cspdarknet_tiny": "https://github.com/yjh0410/image_classification_pytorch/releases/download/weight/cspdarknet_tiny.pth",
     "cspdarknet53": "https://github.com/yjh0410/image_classification_pytorch/releases/download/weight/cspdarknet53_silu.pth",
 }