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'