pytorch加载imagenet的floder数据集

由于kaggle对imagenet的数据集做了整理,直接利用torch.dataset.ImageNet无法加载数据集,

https://github.com/pytorch/examples/tree/main/imagenet

提供了解决办法

posted @ 2024-06-07 13:18  FigureOut  阅读(4)  评论(0编辑  收藏  举报