具有预训练权重的视觉模型的实现
tvmodels的Python项目详细描述
电视型号
tvmodels库包含在imagenet上训练的pytorch中的预训练视觉模型。有些模型在torchvision中提供，但有些没有，因此您可以在此处加载它们。
安装
运行以下命令进行安装：
pipinstalltvmodels
colab
如果在google colab上显示ModuleNotFoundError，请使用以下命令：
!gitclonehttps://github.com/rohitgr7/tvmodels.gitimportsyssys.path.append('/content/tvmodels')
用法
fromtvmodels.modelsimportse_resnet50,resnet18# Load the modelsse_res_model=se_resnet50(pretrained=True)res_model=resnet18(pretrained=True)
可用型号
- [X]resnet（s）
- [X]resnext（s）
- [X]SERESnet（s）
- [X]seresnext（s）
- [X]传感器154
- [X]效率设置
欢迎加入QQ群-->： 979659372
