resnet的预训练权重
时间: 2023-10-22 19:07:40 浏览: 187
经典神经网络resnet101预训练权重文件
5星 · 资源好评率100%
ResNet是一种深度卷积神经网络,它在ImageNet数据集上取得了很好的表现。预训练权重可以从PyTorch官方网站上下载,也可以使用torchvision库中的预训练模型。具体来说,可以使用以下代码加载ResNet50的预训练权重:
```python
import torch
import torchvision.models as models
resnet50 = models.resnet50(pretrained=True)
```
阅读全文