changed the chessboard shape

app.py

@@ -1,17 +1,20 @@
-import numpy as np
 import gradio as gr
 from distortion import main
-import os
+
 
 
 
 
 coords = [[0,0]]
 
-import gradio as gr
-import cv2
 
 
+def get_select_coords(img, evt: gr.SelectData):
+    row, col = evt.index
+    coords.append([row, col])
+    selected_coords.value = f"Selected Coordinates: ({row}, {col})"
+    return coords[-1]
+            
 
 SHARED_UI_WARNING = f'''##### Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed tempus dignissim odio, at elementum erat vulputate sit amet. Vestibulum sodales viverra fermentum. In ac hendrerit dolor, vitae mattis odio. Maecenas suscipit consectetur suscipit. Curabitur sodales dui eget neque venenatis tincidunt. In sed libero mi. Nam sollicitudin metus urna, sit amet sagittis ex laoreet sed.
 
@@ -39,12 +42,6 @@ with gr.Blocks() as demo:
     with gr.Tab("Pick Corners"):
         with gr.Row():
 
-            def get_select_coords(img, evt: gr.SelectData):
-                row, col = evt.index
-                coords.append([row, col])
-                selected_coords.value = f"Selected Coordinates: ({row}, {col})"
-                return coords[-1]
-
             input_img = gr.Image(label="Select Image")
             selected_coords = gr.Textbox(label="Selected Coordinates")
 
@@ -59,9 +56,9 @@ with gr.Blocks() as demo:
                     chess_vert_input = gr.Textbox(label="Enter the number of squares in vertical direction")
                     chess_horz_input = gr.Textbox(label="Enter the number of squares in horizontal direction")
                 download_dxf = gr.Button("Generate the YAML File")
-                yaml_file = gr.File()
-                
+                error_box = gr.Textbox(label="Error", visible=False)
                 
+                yaml_file = gr.File()
                 download_dxf.click(main, inputs=[distorted_image, chess_vert_input, chess_horz_input], outputs= yaml_file)
             with gr.Column():
                 gr.Markdown(SHARED_UI_WARNING)

distortion.py

@@ -3,6 +3,7 @@ import numpy as np
 from pathlib import Path
 
 
+
 def save_coefficients(mtx, dist, path):
     """Save camera matrix and distortion coefficients to file."""
     cv_file = cv2.FileStorage(path, cv2.FILE_STORAGE_WRITE)
@@ -26,7 +27,7 @@ def main(filename, board_vert, board_horz):
     filename_stem = Path(filename).stem
 
     # Define the checkerboard pattern size and criteria for corner detection
-    CHECKERBOARD = (int(board_vert), int(board_horz))
+    CHECKERBOARD = (int(board_vert)-1, int(board_horz)-1)
     #CHECKERBOARD = (9, 7)
     criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 30, 0.001)
 
@@ -55,18 +56,15 @@ def main(filename, board_vert, board_horz):
 
     # Calibrate camera using object points and image points
     h, w = gray.shape[:2]
-    ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(object_points, image_points, gray.shape[::-1], None, None)
-
-    # Save camera matrix and distortion coefficients to file
-    save_coefficients(mtx, dist, f"{filename_stem}.yml")
-
-    # Load camera matrix and distortion coefficients from file
-    mtx, dist = load_coefficients(f'{filename_stem}.yml')
-
-    # Undistort the original image using camera matrix and distortion coefficients
-    original = cv2.imread(filename)
-    dst = cv2.undistort(original, mtx, dist, None, None)
-
-    # Save the undistorted image
-    cv2.imwrite(f'undist_{filename_stem}.jpg', dst)
-    return f"{filename_stem}.yml"
+    try:
+        ret, mtx, dist, rvecs, tvecs = cv2.calibrateCamera(object_points, image_points, gray.shape[::-1], None, None)
+        # Save camera matrix and distortion coefficients to file
+        save_coefficients(mtx, dist, f"{filename_stem}.yml")
+        return f"{filename_stem}.yml"
+    except:
+        print("Please check the Chessboard Dimensions")
+        
+    
+    
+
+    

requirements.txt

@@ -1,2 +1,3 @@
 numpy
-gradio
+gradio
+opencv-python