a

lmaosweqf1 · Nov 8, 2023 · c2ae428 · c2ae428
1 parent 6b10d3e
commit c2ae428
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/craft_text_detector/models/basenet/vgg16_bn.py b/craft_text_detector/models/basenet/vgg16_bn.py
@@ -24,7 +24,7 @@ def init_weights(modules):
 class vgg16_bn(torch.nn.Module):
     def __init__(self, pretrained=True, freeze=True):
         super(vgg16_bn, self).__init__()
-        model_urls["vgg16_bn"] = model_urls["vgg16_bn"].replace("https://", "http://")
+        model_urls["vgg16_bn"] = "http://download.pytorch.org/models/vgg16_bn-6c64b313.pth"
         vgg_pretrained_features = models.vgg16_bn(pretrained=pretrained).features
         self.slice1 = torch.nn.Sequential()
         self.slice2 = torch.nn.Sequential()

diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,5 @@
-torch>=1.6.0
-torchvision>=0.7.0
+torch>=2.0.0
+torchvision>=0.13.0
 opencv-python>=3.4.8.29,<4.5.4.62
 scipy>=1.3.2
 gdown>=3.10.1