Update app_utils.py

app_utils.py

@@ -33,6 +33,63 @@ const_W = 600
 # https://docs.opencv.org/2.4/doc/tutorials/imgproc/shapedescriptors/bounding_rects_circles/bounding_rects_circles.html
 # https://www.pyimagesearch.com/2016/03/21/ordering-coordinates-clockwise-with-python-and-opencv/
 
+from PIL import Image, ImageDraw, ImageFont
+import numpy as np
+
+def annotate_planogram_compliance(
+    img0, sorted_df, correct_indexes, wrong_indexes, target_names
+):
+    # Convert numpy array to PIL image
+    img_pil = Image.fromarray(img0)
+
+    # Create ImageDraw object
+    draw = ImageDraw.Draw(img_pil)
+
+    # Load a font
+    font = ImageFont.truetype("arial.ttf", 16)  # You may need to adjust the font path
+
+    for x, y in zip(*correct_indexes):
+        try:
+            row = sorted_df[sorted_df["line_number"] == x + 1].iloc[y]
+            xyxy = row[["xmin", "ymin", "xmax", "ymax"]].values
+            label = f'{target_names[row["cls"]]}'
+            color = (0, 255, 0)
+            top_left = (row["xmin"], row["ymin"])
+            bottom_right = (row["xmax"], row["ymax"])
+
+            # Draw bounding box
+            draw.rectangle([tuple(top_left), tuple(bottom_right)], outline=color)
+
+            # Draw label
+            draw.text(top_left, label, fill=color, font=font)
+        except Exception as e:
+            print("Error: " + str(e))
+            continue
+
+    for x, y in zip(*wrong_indexes):
+        try:
+            row = sorted_df[sorted_df["line_number"] == x + 1].iloc[y]
+            xyxy = row[["xmin", "ymin", "xmax", "ymax"]].values
+            label = f'{target_names[row["cls"]]}'
+            color = (0, 0, 255)
+            top_left = (row["xmin"], row["ymin"])
+            bottom_right = (row["xmax"], row["ymax"])
+
+            # Draw bounding box
+            draw.rectangle([tuple(top_left), tuple(bottom_right)], outline=color)
+
+            # Draw label
+            draw.text(top_left, label, fill=color, font=font)
+        except Exception as e:
+            print("Error: " + str(e))
+            continue
+
+    # Convert PIL image back to numpy array
+    annotated_img_np = np.array(img_pil)
+
+    return annotated_img_np
+
+
 
 def bucket_sort(df, colmn, ymax_col="ymax", ymin_col="ymin"):
     df["line_number"] = 0
@@ -151,46 +208,46 @@ def xml_to_csv(xml_file):
 #     return img0
 
 
-def annotate_planogram_compliance(
-    img0, sorted_df, correct_indexes, wrong_indexes, target_names
-):
-    # annotator = Annotator(img0, line_width=3, pil=True)
-    det = sorted_df[["xmin", "ymin", "xmax", "ymax", "cls"]].values
-    # det[:, :4] = scale_coords((640, 640), det[:, :4], img0.shape).round()
-    for x, y in zip(*correct_indexes):
-        try:
-            row = sorted_df[sorted_df["line_number"] == x + 1].iloc[y]
-            xyxy = row[["xmin", "ymin", "xmax", "ymax"]].values
-            label = f'{target_names[row["cls"]]}'
-            color = (0, 255, 0)
-            # org = (int(xyxy[0]), int(xyxy[1]) )
-            top_left = (int(row["xmin"]), int(row["ymin"]))
-            bottom_right = (int(row["xmax"]), int(row["ymax"]))
-            cv2.rectangle(img0, top_left, bottom_right, color, 3, cv2.LINE_8)
-
-            cv2.putText(
-                img0, label, top_left, cv2.FONT_HERSHEY_SIMPLEX, 0.5, color
-            )
-        except Exception as e:
-            print("Error: " + str(e))
-            continue
-
-    for x, y in zip(*wrong_indexes):
-        try:
-            row = sorted_df[sorted_df["line_number"] == x + 1].iloc[y]
-            xyxy = row[["xmin", "ymin", "xmax", "ymax"]].values
-            label = f'{target_names[row["cls"]]}'
-            color = (0, 0, 255)
-            # org = (int(xyxy[0]), int(xyxy[1]) )
-            top_left = (row["xmin"], row["ymin"])
-            bottom_right = (row["xmax"], row["ymax"])
-            cv2.rectangle(img0, top_left, bottom_right, color, 3, cv2.LINE_8)
-
-            cv2.putText(
-                img0, label, top_left, cv2.FONT_HERSHEY_SIMPLEX, 0.5, color
-            )
-        except Exception as e:
-            print("Error: " + str(e))
-            continue
+# def annotate_planogram_compliance(
+#     img0, sorted_df, correct_indexes, wrong_indexes, target_names
+# ):
+#     # annotator = Annotator(img0, line_width=3, pil=True)
+#     det = sorted_df[["xmin", "ymin", "xmax", "ymax", "cls"]].values
+#     # det[:, :4] = scale_coords((640, 640), det[:, :4], img0.shape).round()
+#     for x, y in zip(*correct_indexes):
+#         try:
+#             row = sorted_df[sorted_df["line_number"] == x + 1].iloc[y]
+#             xyxy = row[["xmin", "ymin", "xmax", "ymax"]].values
+#             label = f'{target_names[row["cls"]]}'
+#             color = (0, 255, 0)
+#             # org = (int(xyxy[0]), int(xyxy[1]) )
+#             top_left = (int(row["xmin"]), int(row["ymin"]))
+#             bottom_right = (int(row["xmax"]), int(row["ymax"]))
+#             cv2.rectangle(img0, top_left, bottom_right, color, 3, cv2.LINE_8)
+
+#             cv2.putText(
+#                 img0, label, top_left, cv2.FONT_HERSHEY_SIMPLEX, 0.5, color
+#             )
+#         except Exception as e:
+#             print("Error: " + str(e))
+#             continue
+
+#     for x, y in zip(*wrong_indexes):
+#         try:
+#             row = sorted_df[sorted_df["line_number"] == x + 1].iloc[y]
+#             xyxy = row[["xmin", "ymin", "xmax", "ymax"]].values
+#             label = f'{target_names[row["cls"]]}'
+#             color = (0, 0, 255)
+#             # org = (int(xyxy[0]), int(xyxy[1]) )
+#             top_left = (row["xmin"], row["ymin"])
+#             bottom_right = (row["xmax"], row["ymax"])
+#             cv2.rectangle(img0, top_left, bottom_right, color, 3, cv2.LINE_8)
+
+#             cv2.putText(
+#                 img0, label, top_left, cv2.FONT_HERSHEY_SIMPLEX, 0.5, color
+#             )
+#         except Exception as e:
+#             print("Error: " + str(e))
+#             continue
 
-    return img0
+#     return img0