Denoise Images Research

Denoise Images Research

Setup local virtual environment

Create and activate virtual environment:

python3 -m venv .venv
source .venv/bin/activate

Install dependencies:

pip install -r requirements.txt

Deactivate virtual environment:

deactivate

Run scripts

Using Tensorflow Docker image with NVidia GPU support

Build the Docker image:

./build_docker.sh

Start the Docker container:

./start_docker.sh

Execute the train.py script inside Docker container:

cd /tmp
python3 train.py

For expermiments, please use the command line arguments:

cd /tmp
python3 train.py --learning-rate=1e-6 --model-name='denoise_unet.h5'