Update utils.py (#233)

v7labs · Oct 21, 2021 · 435331f · 435331f
1 parent 30ee564
commit 435331f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/darwin/torch/utils.py b/darwin/torch/utils.py
@@ -100,7 +100,9 @@ def detectron2_register_dataset(
     catalog_name = f"darwin_{dataset_path.name}"
     if partition:
         catalog_name += f"_{partition}"
-    classes = get_classes(dataset_path, annotation_type="polygon")
+
+    classes = get_classes(dataset_path=dataset_path, release_name=release_name, annotation_type="polygon")
+
     DatasetCatalog.register(
         catalog_name,
         lambda partition=partition: list(