训练自己的网络时,预训练模型采用ImageNet-1K好还是ImageNet-21K更好
时间: 2024-01-05 22:03:24 浏览: 332
kears下imageNet数据集的预训练模型
预训练模型采用ImageNet-21K更好。因为ImageNet-1K只包含1000个类别,而ImageNet-21K包含超过2万个类别,因此更具有代表性和广泛性。此外,使用更多的类别进行预训练可以提高模型的泛化能力和适应性,这对于应用在不同领域的任务中非常重要。但需要注意的是,使用ImageNet-21K进行预训练需要更多的计算资源和时间。
阅读全文