From 8eefddc0c8b270c4ed5b43d8b0d4f0789e04d157 Mon Sep 17 00:00:00 2001 From: onb1259 <onb1259@onb.ac.at> Date: Mon, 14 Aug 2023 14:33:16 +0200 Subject: [PATCH] added requirements file --- .gitignore | 1 + requirements.txt | 6 ++++++ train_colorizer.ipynb | 2 +- 3 files changed, 8 insertions(+), 1 deletion(-) create mode 100644 requirements.txt diff --git a/.gitignore b/.gitignore index 0843976..7368bbb 100644 --- a/.gitignore +++ b/.gitignore @@ -4,3 +4,4 @@ __pycache__ *.png .ipynb_checkpoints postcards_256/* +venv \ No newline at end of file diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..97bf5f5 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,6 @@ +torch +scikit-image +matplotlib +numpy +pillow +requests diff --git a/train_colorizer.ipynb b/train_colorizer.ipynb index 95662f1..d2bc462 100644 --- a/train_colorizer.ipynb +++ b/train_colorizer.ipynb @@ -17,7 +17,6 @@ "source": [ "import os\n", "import requests\n", - "import torch\n", "from torch.utils.data import random_split\n", "from torch.utils.data import DataLoader\n", "from models.ColDataset import ColDataset\n", @@ -171,6 +170,7 @@ ], "source": [ "# Get a random AKON postcard\n", + "import torch\n", "from random import randint\n", "from skimage import io\n", "from iiif_utils import get_manifesturls_from_collectionurl, get_imgurls_from_manifesturl\n", -- GitLab