Github项目pytorch-image-models，整合图像模型、脚本和预训练参数

【导读】Github项目pytorch-image-models整合了PyTorch的许多图像模型、脚本和预训练参数。除了来自Torchvision等现有的仓库的模型，作者自己也提供了一些模型。

Github项目rwightman/pytorch-image-models整合了PyTorch的许多图像模型、脚本和预训练参数。项目中的模型主要有下面三个来源：

• Torchvision

  https://github.com/pytorch/vision/tree/master/torchvision/models

• Cadene's Pretrained Models

  https://github.com/Cadene/pretrained-models.pytorch

• 作者自己提供：

  https://github.com/rwightman/pytorch-dpn-pretrained

目前pytorch-image-models整合的模型大类如下：

• ResNet/ResNeXt
  

• DenseNet

• Squeeze-and-Excitation ResNet/ResNeXt

• Inception-ResNet-V2 and Inception-V4

• Xeption
  

• PNasNet and NASNet-A
  

• DPN

• Generic EfficientNet

具体模型列表可以参考：

• https://github.com/rwightman/pytorch-image-models#models

目前pytorch-image-models项目已经获得了900+ Stars：

参考链接：

• https://github.com/rwightman/pytorch-image-models