Dataset文件下载以及使用,以nuswide为例

文章目录

  • 文件夹结构
  • 如何使用
    • 继承torch.utils.data.Dataset构建新的class
    • 构建新的Dataloader

数据集要求以文章 multi-label learning from single positive label为例;

文件夹结构

  • Flickr/
  • formatted_train_images.npy
  • formatted_train_labels.npy
  • formatted_train_labels_obs.npy
  • formatted_val_images.npy
  • formatted_val_labels.npy
  • formatted_val_labels_obs.npy
  • val_features_imagenet_resnet50.npy
  • train_features_imagenet_resnet50.npy

我是这么放置的,其中含有obs的文件是通过运行文件夹preproc下的generate_observed_labels.py运行之后生成的;

如何使用

继承torch.utils.data.Dataset构建新的class

构建新的Dataloader

你可能感兴趣的:(DeepLearning,深度学习,人工智能)