search

ivadomed / utilities

Repository containing utility scripts for handling various aspects in a DL model training pipeline
MIT License
8 stars 0 forks source link

Plot nnUNet training log #41

Closed valosekj closed 7 months ago

valosekj commented 7 months ago

This PR adds a script to plot epoch number vs Pseudo dice based on a nnUNet training log file, separately for each class.

Usage:

# Create venv and install dependencies
python3 -m venv venv
source ./venv/bin/activate
pip install pandas plotly kaleido
# Run the script and save the figure as PNG
python plot_nnunet_training_log.py -i <path_to_log_file>
# Run the script, save the figure as PNG and show interactive figure
python plot_nnunet_training_log.py -i <path_to_log_file> -interactive-figure
image
valosekj commented 7 months ago

Good catch @Nilser3! Thank you! I updated dependencies in my original comment.

valosekj commented 7 months ago

Thanks for the reviews!

The figure now looks like:

image