fix function names

Author: Matt Sokoloff

examples/model_assisted_labeling/image_mal.ipynb

@@ -10,7 +10,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": 1,
    "id": "committed-richards",
    "metadata": {},
    "outputs": [],
@@ -31,10 +31,10 @@
     "import ndjson\n",
     "from image_model import predict, class_mappings, load_model\n",
     "from image_mal_utils import (\n",
-    "    visualize_bbox_upload,\n",
-    "    visualize_poly_upload,\n",
-    "    visualize_point_upload,\n",
-    "    visualize_mask_uploads\n",
+    "    visualize_bbox_ndjsons,\n",
+    "    visualize_poly_ndjsons,\n",
+    "    visualize_point_ndjsons,\n",
+    "    visualize_mask_ndjsons\n",
     ")"
    ]
   },
@@ -319,10 +319,10 @@
    ],
    "source": [
     "image = np.array(Image.open(BytesIO(np_image_bytes[0])))\n",
-    "image = visualize_bbox_upload(image,nd_box_payloads ,(0,255,0))\n",
-    "image = visualize_poly_upload(image,nd_poly_payloads ,(0,0,255))\n",
-    "image = visualize_point_upload(image,nd_point_payloads ,(0,255,255))\n",
-    "image = visualize_mask_uploads(image, nd_mask_payloads)\n",
+    "image = visualize_bbox_ndjsons(image,nd_box_payloads ,(0,255,0))\n",
+    "image = visualize_poly_ndjsons(image,nd_poly_payloads ,(0,0,255))\n",
+    "image = visualize_point_ndjsons(image,nd_point_payloads ,(0,255,255))\n",
+    "image = visualize_mask_ndjsons(image, nd_mask_payloads)\n",
     "Image.fromarray(image)"
    ]
   },

examples/model_assisted_labeling/image_mal_utils.py

@@ -4,28 +4,28 @@
 from io import BytesIO
 from PIL import Image
 
-def visualize_bbox_upload(image, bbox_ndjsons, color):
+def visualize_bbox_ndjsons(image, bbox_ndjsons, color):
     for bbox_ndjson in bbox_ndjsons:
         bbox = bbox_ndjson['bbox']
         start_point = (int(bbox['left']), int(bbox['top']))
         end_point = (int(bbox['left'] + bbox['width']), int(bbox['top'] + bbox['height']))
         image = cv2.rectangle(image, start_point, end_point, thickness = 2, color = color)
     return image
 
-def visualize_poly_upload(image, poly_ndjsons, color):
+def visualize_poly_ndjsons(image, poly_ndjsons, color):
     for poly_ndjson in poly_ndjsons:
         pts = [[poly['x'], poly['y']] for poly in poly_ndjson['polygon']]
         pts = np.array(pts).astype(np.int32)
         image = cv2.polylines(image,[pts],True,color, thickness = 2)
     return image
 
-def visualize_point_upload(image, point_ndjsons, color):
+def visualize_point_ndjsons(image, point_ndjsons, color):
     for point_ndjson in point_ndjsons:
         point = point_ndjson['point']
         image = cv2.circle(image, (point['x'],point['y']), radius=10, color=color, thickness=-1)
     return image
 
-def visualize_mask_uploads(image, mask_ndjsons):
+def visualize_mask_ndjsons(image, mask_ndjsons):
     masks = []
     for ndjson in mask_ndjsons:
         instanceURI = ndjson['mask']['instanceURI']