update dataset in measures

mu_map/eval/measures.py

@@ -116,7 +116,7 @@ if __name__ == "__main__":
     parser.add_argument(
         "--dataset_dir",
         type=str,
-        default="data/initial/",
+        default="data/second/",
         help="directory where the dataset is found",
     )
     parser.add_argument(
@@ -147,7 +147,8 @@ if __name__ == "__main__":
     torch.set_grad_enabled(False)
 
     device = torch.device(args.device)
-    model = UNet(features=[32, 64, 128, 256, 512])
+    # model = UNet(features=[32, 64, 128, 256, 512])
+    model = UNet(features=[64, 128, 256, 512])
     model.load_state_dict(torch.load(args.weights, map_location=device))
     model = model.to(device).eval()