This is a graduate project for my undergraduate study.
It aims to build an app that can be used to colour and repair images. Also, it includes some basic functions that an image editor will do.
See the brief intro video via Bilibili in Chinese or YouTube in English.
By default, this project uses PyTorch packages with CPU versions, just in case you want to make it back to the CPU version, run the following command line:
pip install --force-reinstall torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cpu
For GPU development, reinstall them by running:
pip install --force-reinstall torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
The GPU version was tested successfully on NVIDIA GTX 1660 SUPER.
Look at pytorch official website for more detail.
Note that package pillow may be reinstalled with a new version
which can lead to an error while launching the app
after changing the torch version. To stop it from happening,
use the following command line to reinstall pillow:
pip install --force-reinstall pillow==9.5.0