[experiment] Add saliency to experiment script

By default, performs sal gen and sal view. Default parameters
are used for now. Steps requiring databases with human annotations
are not yet included in the experiment script.

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")