【导读】Github项目pytorch-image-models整合了PyTorch的许多图像模型、脚本和预训练参数。除了来自Torchvision等现有的仓库的模型,作者自己也提供了一些模型。 Github项目rwightman/pytorch-image-models整合了PyTorch的许多图像模型、脚本和预训练参数。项目中的模型主要有下面三个来源:
目前pytorch-image-models整合的模型大类如下:
具体模型列表可以参考:
目前pytorch-image-models项目已经获得了900+ Stars: 参考链接:
|
|
来自: LibraryPKU > 《机器学习》