文章目录
- 文件夹结构
- 如何使用
- 继承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
运行之后生成的;