/rescuenet

Primary LanguageJupyter NotebookMIT LicenseMIT

rescuenet

Contribution to the RescueNet Challenge 2023.

Run training with python fine_tune.py. Will use config/att_unet.json. Hyperparameters can be changed by writing a new config in the same style. For more information on possible values see config/README.md. Logs will be stored in metrics/scores/fine_tune. Adding --use_wandb option to train call will additionally use wandb for monitoring.

Hyperparameter tuning and prediction scripts are not tested yet.