Skip to content

Commit 64862b5

Browse files
ducha-aikifmassa
authored andcommitted
Added urls to Harris part of Phototour (#336)
* Added urls to Harris patches on Phototour * fixed formatting * fixed flake8 complains
1 parent c31c3d7 commit 64862b5

File tree

1 file changed

+21
-5
lines changed

1 file changed

+21
-5
lines changed

torchvision/datasets/phototour.py

Lines changed: 21 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,21 @@ class PhotoTour(data.Dataset):
2424
2525
"""
2626
urls = {
27+
'notredame_harris': [
28+
'http://matthewalunbrown.com/patchdata/notredame_harris.zip',
29+
'notredame_harris.zip',
30+
'69f8c90f78e171349abdf0307afefe4d'
31+
],
32+
'yosemite_harris': [
33+
'http://matthewalunbrown.com/patchdata/yosemite_harris.zip',
34+
'yosemite_harris.zip',
35+
'a73253d1c6fbd3ba2613c45065c00d46'
36+
],
37+
'liberty_harris': [
38+
'http://matthewalunbrown.com/patchdata/liberty_harris.zip',
39+
'liberty_harris.zip',
40+
'c731fcfb3abb4091110d0ae8c7ba182c'
41+
],
2742
'notredame': [
2843
'http://icvl.ee.ic.ac.uk/vbalnt/notredame.zip',
2944
'notredame.zip',
@@ -40,10 +55,12 @@ class PhotoTour(data.Dataset):
4055
'fdd9152f138ea5ef2091746689176414'
4156
],
4257
}
43-
mean = {'notredame': 0.4854, 'yosemite': 0.4844, 'liberty': 0.4437}
44-
std = {'notredame': 0.1864, 'yosemite': 0.1818, 'liberty': 0.2019}
45-
lens = {'notredame': 468159, 'yosemite': 633587, 'liberty': 450092}
46-
58+
mean = {'notredame': 0.4854, 'yosemite': 0.4844, 'liberty': 0.4437,
59+
'notredame_harris': 0.4854, 'yosemite_harris': 0.4844, 'liberty_harris': 0.4437}
60+
std = {'notredame': 0.1864, 'yosemite': 0.1818, 'liberty': 0.2019,
61+
'notredame_harris': 0.1864, 'yosemite_harris': 0.1818, 'liberty_harris': 0.2019}
62+
lens = {'notredame': 468159, 'yosemite': 633587, 'liberty': 450092,
63+
'liberty_harris': 379587, 'yosemite_harris': 450912, 'notredame_harris': 325295}
4764
image_ext = 'bmp'
4865
info_file = 'info.txt'
4966
matches_files = 'm50_100000_100000_0.txt'
@@ -57,7 +74,6 @@ def __init__(self, root, name, train=True, transform=None, download=False):
5774

5875
self.train = train
5976
self.transform = transform
60-
6177
self.mean = self.mean[name]
6278
self.std = self.std[name]
6379

0 commit comments

Comments
 (0)