def phototour(): return itertools.chain(*[ collect_download_configs( lambda: datasets.PhotoTour(ROOT, name=name, download=True), name=f"PhotoTour, {name}", file="phototour", ) # The names postfixed with '_harris' point to the domain 'matthewalunbrown.com'. For some reason all # requests timeout from within CI. They are disabled until this is resolved. for name in ("notredame", "yosemite", "liberty") # "notredame_harris", "yosemite_harris", "liberty_harris" ])
download=True) stl10_test = datasets.STL10(root=os.path.join(dataset_path, "stl10"), split="test", download=True) stl10_unlabeled = datasets.STL10(root=os.path.join(dataset_path, "stl10"), split="unlabeled", download=True) print("STL10数据集,train data shape:", stl10_train.data.shape) print("STL10数据集,train labels shape:", stl10_train.labels.shape) print("STL10数据集,test data shape:", stl10_test.data.shape) print("STL10数据集,test labels shape:", stl10_test.labels.shape) print("STL10数据集,unlabeled data shape:", stl10_unlabeled.data.shape) print("STL10数据集,unlabeled labels shape:", stl10_unlabeled.labels.shape) '''PhotoTour数据集''' phototour_train = datasets.PhotoTour(root=os.path.join(dataset_path, "phototour"), name="phototour_train", train=True, download=True) phototour_test = datasets.PhotoTour(root=os.path.join(dataset_path, "phototour"), name="phototour_test", train=False, download=True) print("PhotoTour数据集,train data shape:", phototour_train.data.shape) print("PhotoTour数据集,train labels shape:", phototour_train.labels.shape) print("PhotoTour数据集,train matches shape:", phototour_train.matches.shape) print("PhotoTour数据集,test data shape:", phototour_test.data.shape) print("PhotoTour数据集,test labels shape:", phototour_test.labels.shape) print("PhotoTour数据集,test matches shape:", phototour_test.matches.shape)