torch.hub.load 下载 EAST 模型
时间: 2023-07-18 09:26:34 浏览: 196
浅谈pytorch中torch.max和F.softmax函数的维度解释
好的,您可以使用以下代码下载EAST模型:
```python
import torch
# 下载EAST模型
model = torch.hub.load('argman/EAST', 'east_resnet50', pretrained='True')
# 打印模型结构
print(model)
```
请确保您安装了`torch`库,并且联网状态良好。这个过程可能需要一些时间来下载模型参数和其他依赖项,具体时间取决于您的网络速度。
阅读全文