diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..ba0430d --- /dev/null +++ b/.gitignore @@ -0,0 +1 @@ +__pycache__/ \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 2b00a8e..5c49a36 100644 --- a/requirements.txt +++ b/requirements.txt @@ -9,5 +9,8 @@ numpy>=1.16.0,<2.0 Pillow==5.3.0 seaborn scipy -tensorflow-gpu==1.15.2 +tensorflow-gpu==1.15 tqdm +lmbd +requests +libml diff --git a/scripts/create_datasets.py b/scripts/create_datasets.py index ad1ea32..3190d0c 100755 --- a/scripts/create_datasets.py +++ b/scripts/create_datasets.py @@ -202,7 +202,7 @@ def download(url, filename): def _load_lsun(category): data = {} for subset in ('train', 'val'): - url = 'http://lsun.cs.princeton.edu/htbin/download.cgi?tag=latest&category=%s&set=%s' % (category, subset) + url = 'http://dl.yf.io/lsun/scenes/{category}_{subset}_lmdb.zip'.format(**locals()) with tempfile.NamedTemporaryFile(delete=False) as f: download(url, f.name) data[subset] = f.name