diff --git a/src/mednet/scripts/experiment.py b/src/mednet/scripts/experiment.py index 38d88fd94c32fee58c8959ea3ed767c59ff0fd2f..275bc8b8bb91417a6467bcc62b4b7af708253533 100644 --- a/src/mednet/scripts/experiment.py +++ b/src/mednet/scripts/experiment.py @@ -67,7 +67,6 @@ def experiment( save_sh_command(output_folder / "command.sh") - # training logger.info("Started training") from .train import train @@ -139,3 +138,35 @@ def experiment( ) logger.info("Ended evaluating") + + logger.info("Started generating saliencies") + + from .saliency.generate import generate + + saliencies_gen_folder = output_folder / "gradcam" / "saliencies" + + ctx.invoke( + generate, + model=model, + datamodule=datamodule, + weight=train_output_folder, + output_folder=saliencies_gen_folder, + ) + + logger.info("Ended generating saliencies") + + logger.info("Started viewing saliencies") + + from .saliency.view import view + + saliencies_view_folder = output_folder / "gradcam" / "visualizations" + + ctx.invoke( + view, + model=model, + datamodule=datamodule, + input_folder=saliencies_gen_folder, + output_folder=saliencies_view_folder, + ) + + logger.info("Ended viewing saliencies")