diff --git a/.gitignore b/.gitignore index 0843976c6a7ed58d7d43fae5ed3746b8eed29783..7368bbbb54b8877fbce97bab543a2f1069fa20e9 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 0000000000000000000000000000000000000000..97bf5f5000457b37f9a43069be9d776eb5c2fb4d --- /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 95662f1a247e705d18d02d0c93ebee4ef630b80e..d2bc4626acb0f996d9de3f445b69be0804d939ab 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",