1. torchvision的models
里面包含常用的深度模型:AlexNet、VGG、GoogleNet等
使用: model=models.resnet50()
2. pretrainedmodels
里面包含常用的深度模型:AlexNet、VGG、GoogleNet等,还包括一些新的模型Xception、SENet等
使用:args1=pretrainedmodels.__dict__['resnet50'](pretrained='imagenet').state_dict()
To be continue...

本文介绍了两个常用的深度学习模型库:torchvision.models 和 pretrainedmodels,它们分别提供了如AlexNet、VGG、GoogleNet等经典模型及Xception、SENet等新模型的预训练版本,便于快速上手和应用。
1653

被折叠的 条评论
为什么被折叠?



