add help and allow setting number of unet features in measures script

mu_map/eval/measures.py

@@ -111,7 +111,8 @@ if __name__ == "__main__":
         help="the model weights which should be scored",
     )
     parser.add_argument("--out", type=str, help="write results as a csv file")
-    parser.add_argument("--scatter_corrected", action="store_true")
+    parser.add_argument("--scatter_corrected", action="store_true", help="load scatter corrected reconstructions")
+    parser.add_argument("--features", type=int, nargs="+", default=[64, 128, 256, 512], help="features defining the U-Net model for evaluation")
 
     parser.add_argument(
         "--dataset_dir",
@@ -147,8 +148,7 @@ if __name__ == "__main__":
     torch.set_grad_enabled(False)
 
     device = torch.device(args.device)
-    # model = UNet(features=[32, 64, 128, 256, 512])
-    model = UNet(features=[64, 128, 256, 512])
+    model = UNet(features=args.features)
     model.load_state_dict(torch.load(args.weights, map_location=device))
     model = model.to(device).eval()