PyTorch图像模型库timm解析
参考文档:0、1、2
简介
PyTorch Image Models (timm)是Ross Wightman创建的深度学习库,是一个大型集合,包括了SOTA计算机视觉模型、神经网络层、实用函数、优化器、调度器、数据加载器、数据增强器以及训练/验证脚本等。
安装
1
pip install timm
示例数据集(可选)
在演示之前,先下载一些流行的数据集作为示范。在这里,Chris Hughes使用了两个数据集:
* 牛津大学IIIT宠物数据集,该数据集有37个类别,每个类别大约有200张图片
* Imagenette,这是Imagenet中10个容易分类的类别的一个子集。