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)
