1

我正在尝试对 CUB-200-2011 数据集进行细粒度的图像分类,但无法找出加载数据的正确方法。我正在使用预训练的 ImageNet ResNet-101 进行训练,方法是冻结除最后两层之外的所有层。建议一种在 Pytorch 中加载数据的方法。

4

0 回答 0