Skip to content

Commit

Permalink
Update data download links.
Browse files Browse the repository at this point in the history
  • Loading branch information
tibuch committed Nov 4, 2021
1 parent 3350a38 commit a6bf7e4
Show file tree
Hide file tree
Showing 6 changed files with 117 additions and 76 deletions.
32 changes: 19 additions & 13 deletions examples/Super-Resolution - CelebA Example.ipynb

Large diffs are not rendered by default.

33 changes: 22 additions & 11 deletions examples/Super-Resolution - MNIST Example.ipynb

Large diffs are not rendered by default.

58 changes: 34 additions & 24 deletions examples/Tomographic Reconstruction - Kanji Example.ipynb

Large diffs are not rendered by default.

62 changes: 36 additions & 26 deletions examples/Tomographic Reconstruction - LoDoPaB Example.ipynb

Large diffs are not rendered by default.

4 changes: 3 additions & 1 deletion fit/datamodules/super_res/SResDataModule.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
import numpy as np
import torch
import wget
import ssl
ssl._create_default_https_context = ssl._create_unverified_context
from pytorch_lightning import LightningDataModule
from torch.utils.data import DataLoader
from torchvision.datasets import MNIST
Expand Down Expand Up @@ -107,7 +109,7 @@ def __init__(self, root_dir, batch_size):

def prepare_data(self, *args, **kwargs):
if not exists(join(self.root_dir, 'gt_data.npz')):
wget.download('https://cloud.mpi-cbg.de/index.php/s/Wtuy9IqUsSpjKav/download',
wget.download('https://download.fht.org/jug/fit/CelebA_SRes_gt_data.npz',
out=join(self.root_dir, 'gt_data.npz'))

gt_data = np.load(join(self.root_dir, 'gt_data.npz'))
Expand Down
4 changes: 3 additions & 1 deletion fit/datamodules/tomo_rec/TRecDataModule.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,8 @@
from fit.utils.utils import normalize

import wget
import ssl
ssl._create_default_https_context = ssl._create_unverified_context


class TomoFITDataModule(LightningDataModule):
Expand Down Expand Up @@ -273,7 +275,7 @@ def __init__(self, root_dir, batch_size, num_angles=33):

def prepare_data(self, *args, **kwargs):
if not exists(join(self.root_dir, 'gt_data.npz')):
wget.download('https://cloud.mpi-cbg.de/index.php/s/7MK9vNUnq4Ndkhg/download',
wget.download('https://download.fht.org/jug/fit/Kanji_TRec_gt_data.npz',
out=join(self.root_dir, 'gt_data.npz'))

gt_data = np.load(join(self.root_dir, 'gt_data.npz'))
Expand Down

0 comments on commit a6bf7e4

Please sign in to comment.