Merge pull request #8 from mo-mo-666/develop

v1.1.0

src/align_images.py

@@ -90,9 +90,12 @@ def __find_one_marker(
         max_area = areas[max_idx]
         # calculate center of gravity
         M = cv2.moments(max_cnt)
-        cx = M["m10"] / M["m00"]
-        cy = M["m01"] / M["m00"]
-        return (cx, cy), max_area
+        try:
+            cx = M["m10"] / M["m00"]
+            cy = M["m01"] / M["m00"]
+            return (cx, cy), max_area
+        except ZeroDivisionError:
+            return (0, 0), 0
 
     def _find_markers(self, binary_img: np.ndarray) -> np.ndarray:
         """

src/pipeline.py

@@ -41,7 +41,7 @@ def pipeline(
     # read metadata
     # metadata = read_metadata(metadata_path, pt2px = dpi[0])
     baseimg, dpi = read_image(baseimg_path)
-    metadata = read_metadata(None, pt2px=dpi[0])
+    metadata = read_metadata(metadata_path, pt2px=dpi[0])
     resize_ratio: float = metadata["resize_ratio"]
     is_align: bool = metadata["is_align"]
     is_marksheet: bool = metadata["is_marksheet"]

src/setting_io_ds.py

@@ -52,10 +52,12 @@ def read_metadata(
     metadata = {}
     if filepath:
         wb = load_workbook(filepath, read_only=True)
-        ws = wb[excel_sheet_name]
-        for row in ws.iter_rows(min_row=3):
-            key, value = row[0].value, row[1].value
-            metadata[key] = value
+        if excel_sheet_name in wb:
+            logger.debug(f"We found the sheet {excel_sheet_name}.")
+            ws = wb[excel_sheet_name]
+            for row in ws.iter_rows(min_row=3):
+                key, value = row[0].value, row[1].value
+                metadata[key] = value
 
     # put default value
     for key, value in SETTING_KEYS_DEFAULT.items():