Merge pull request #2 from mayrajeo/cocores

more migrating fixes
mayrajeo · Sep 19, 2023 · bfdea1b · bfdea1b
2 parents 95da519 + 5cd0361
commit bfdea1b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/geo2ml/data/cv.py b/geo2ml/data/cv.py
@@ -285,6 +285,7 @@ def shp_to_coco_results(
     raster_path: Path,
     coco_dict: Path,
     outpath: Path,
+    label_col: str = "label_id",
     rotated_bbox: bool = False,
 ):
     "Convert vector predictions into coco result format to be fed into COCO evaluator"

diff --git a/nbs/13_data.cv.ipynb b/nbs/13_data.cv.ipynb
@@ -603,7 +603,8 @@
    "source": [
     "#| export\n",
     "\n",
-    "def shp_to_coco_results(prediction_path:Path, raster_path:Path, coco_dict:Path, outpath:Path, rotated_bbox:bool=False):\n",
+    "def shp_to_coco_results(prediction_path:Path, raster_path:Path, coco_dict:Path, outpath:Path, \n",
+    "                        label_col:str='label_id', rotated_bbox:bool=False):\n",
     "    \"Convert vector predictions into coco result format to be fed into COCO evaluator\"\n",
     "\n",
     "    with open(coco_dict) as f:\n",