In "unet_baseline" folder run :

python3 training.py 
python3 inference.py 

Please specify parameters