Optimize evaluation

pyproject.toml

@@ -49,6 +49,8 @@ dependencies = [
   "pyreadline3; sys_platform == 'win32'",
   "cellmap-flow@git+https://github.com/janelia-cellmap/cellmap-flow@1ece404",
   "pykdtree",
+  "fastremap",
+  "connected-components-3d",
   ]
 
 [project.optional-dependencies]

src/cellmap_segmentation_challenge/cli/evaluate.py

@@ -1,8 +1,8 @@
 import os
 import click
 
-from cellmap_segmentation_challenge.evaluate import (
-    INSTANCE_CLASSES,
+from cellmap_segmentation_challenge.evaluate import INSTANCE_CLASSES
+from cellmap_segmentation_challenge import (
     PROCESSED_PATH,
     SUBMISSION_PATH,
     TRUTH_PATH,

src/cellmap_segmentation_challenge/cli/fetch_data.py

@@ -441,3 +441,4 @@ def fetch_data_cli(
         log.unbind("save_location")
         log.info(f"Done after {time.time() - fetch_save_start:0.3f}s")
         log.info(f"Data saved to {dest_path_abs}")
+        pool.shutdown(wait=True)