more migrating fixes

mayrajeo · Sep 19, 2023 · 5cd0361 · 5cd0361
1 parent 2babb6f
commit 5cd0361
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",