From 2c470c771fa392b102b0dc57036392654754a3b4 Mon Sep 17 00:00:00 2001
From: Bubbliiiing <3323290568@qq.com>
Date: Fri, 8 Jul 2022 12:43:10 +0800
Subject: [PATCH] update pretrained

---
 nets/CSPdarknet.py | 4 ++--
 utils/utils.py     | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/nets/CSPdarknet.py b/nets/CSPdarknet.py
index 0d4dc69..91cabed 100644
--- a/nets/CSPdarknet.py
+++ b/nets/CSPdarknet.py
@@ -109,11 +109,11 @@ class CSPDarknet(nn.Module):
         if pretrained:
             phi = 'l'
             url = {
-                "l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/cspdarknet_backbone_l.pth',
+                "l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/cspdarknet_backbone.pth',
             }[phi]
             checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu", model_dir="./model_data")
             self.load_state_dict(checkpoint, strict=False)
-            print("Load weights from ", url.split('/')[-1])
+            print("Load weights from " + url.split('/')[-1])
 
     def forward(self, x):
         x = self.stem(x)
diff --git a/utils/utils.py b/utils/utils.py
index 57db181..4c22c2c 100644
--- a/utils/utils.py
+++ b/utils/utils.py
@@ -77,7 +77,7 @@ def download_weights(model_dir="./model_data"):
     
     phi = "l"
     download_urls = {
-        "l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/cspdarknet_backbone_l.pth',
+        "l" : 'https://github.com/bubbliiiing/yolov7-pytorch/releases/download/v1.0/cspdarknet_backbone.pth',
     }
     url = download_urls[phi]
     
-- 
GitLab