文章目录
- 文件夹结构
- 如何使用
- 继承torch.utils.data.Dataset构建新的class
- 构建新的Dataloader
数据集要求以文章 multi-label learning from single positive label为例;
文件夹结构
Flickr/formatted_train_images.npyformatted_train_labels.npyformatted_train_labels_obs.npyformatted_val_images.npyformatted_val_labels.npyformatted_val_labels_obs.npyval_features_imagenet_resnet50.npytrain_features_imagenet_resnet50.npy
我是这么放置的,其中含有obs的文件是通过运行文件夹preproc下的generate_observed_labels.py运行之后生成的;