pytorch自带有一些高级的复杂模型,我们可以通过 torchvision.models 调用,例如torchvision.models.densenet169(pretrained=True)就调用了densenet169的预训练模型。

先来看看这个模块涵盖了哪些模型:
https://pytorch.org/vision/stable/models.html获取详细信息。

注意

基本所有预训练模型都是基于ImageNet数据集训练的,得先从从网站(https://s3.amazonaws.com/deep-learning-models/imag