From 7414ad5d51cd73683487babfc8d6c7767ba2abd8 Mon Sep 17 00:00:00 2001 From: Philip Meixner Date: Sat, 9 Dec 2023 17:52:35 +0100 Subject: [PATCH] update readme --- README.md | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/README.md b/README.md index 2ebd7cb..eae789d 100644 --- a/README.md +++ b/README.md @@ -1,7 +1,12 @@ # Image depixelation project -Training a model which's task is to recreate a pixelated area of an Image - -In this repository you can find a project I did in university last semester, where the aim was to train a model which can recreate a certain area of an image which was pixelated before being fed into the model. +In this repository you can find a project I did in university in the summer term 2022, where the aim was to train a model which can recreate a certain area of an image which was pixelated before being fed into the model. ![Image of the task](task.png "Task") + +The dataset which was used to train the model consisted of 100 images per student, which added up to about 35 000 images. + +Using image augmentation I doubled the amount of images using a little script, which can be found in "image_augmentation.py". Preprocessing of the images was handled by the "prep_image.py", "stack_with_padding.py" and "create_dataset.py" were used to bring the images into datasets which were suitable for PyTorch Dataloaders. + +In the file "model.py" you can find my final model architecture and some prior attempts. The final model architecture is definetely simpler than the others, simply because the more complex models did not perform well due to bad generalization. +