分类火炬的技巧 使用pytorch进行图像分类教程的技巧的载体。 基于使用自定义数据集实现分类代码库。 作者:hoya012 最后更新:2020.08.06 0.实验设置(我使用了1个GTX 1080 Ti GPU!) 0-1。 准备图书馆 pip install - r requirements . txt 0-2。 下载数据集(Kaggle英特尔映像分类) 该数据包含大约25,000张大小为150x150的图像,分布在6个类别中。 {'建筑物'-> 0,'森林'-> 1,'冰川'-> 2,'山'-> 3,'海'-> 4,'街道'-> 5} 0-3。 下载ImageNet预训练的权重(EfficientNet,RegNet) 下载RegNetY-1.6GF和EfficientNet-B2砝码 1.基线培训设置 来自Torchvision.models的ImageNet预训练的R