Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -109,7 +109,7 @@ def calculate_midpoint(top_left, bottom_right):
     ym = int((y1 + y2) / 2)
     return (xm, ym)
 
-def mid_points_bbox(bbox):
+def mid_points_bbox(bbox, ratio):
   midpoints = []
   for i in range(len(bbox)):
     x1 = int(bbox[i][0])
@@ -119,6 +119,7 @@ def mid_points_bbox(bbox):
     top_left_corner = (x1, y1)
     bottom_right_corner = (x2, y2)
     door_width = calculate_width(bbox[i])
+    door_width = door_width * ratio
     midpoint = calculate_midpoint(top_left_corner, bottom_right_corner)
     midpoints.append((midpoint, door_width))
   return midpoints
@@ -127,7 +128,7 @@ def create_annotations(door_kind, midpoints):
   door = door_kind
   annotations = []
   for i in range(len(midpoints)):
-    annotations.append((midpoints[i][0][0],midpoints[i][0][1], door+f" with {midpoints[i][1]} pixels width"))
+    annotations.append((midpoints[i][0][0],midpoints[i][0][1], door+f" with {midpoints[i][1]} meters width"))
   return annotations
 
 def add_annotations_to_pdf(image, pdf_name, annotation_s, annotation_d):
@@ -168,10 +169,10 @@ def add_annotations_to_pdf(image, pdf_name, annotation_s, annotation_d):
 
 def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath): ####pdf_fullpath here is the data and not the path 
     img_pillow = convert2pillow(pdf_fullpath)
-    new_image = img_pillow.resize((2384, 1684))
+    #new_image = img_pillow.resize((2384, 1684))
     
     # Specify the number of classes (including the background)
-    num_classes = 6  # Ensure this matches the saved model's number of classes
+    num_classes = 3  # Ensure this matches the saved model's number of classes
     
     # Load the model with the specified number of classes
     model = get_model(num_classes)
@@ -192,15 +193,15 @@ def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath): ####pdf_fullpath her
     model.to(device)
     
     # START INFERENCE
-    sbox, dbox = ev_model(new_image, model, device, 0.6)
+    sbox, dbox = ev_model(img_pillow, model, device, 0.6)
     
     #Dataframe for Doors count
     doors_count = {'Type': ['Single Doors', 'Double Doors'], 'Quantity': [len(sbox), len(dbox)]}
     df_doors = pd.DataFrame(doors_count)
 
     #Door's details for annotation
-    single_info = mid_points_bbox(sbox)
-    double_info = mid_points_bbox(dbox)
+    single_info = mid_points_bbox(sbox, ratio)
+    double_info = mid_points_bbox(dbox, ratio)
 
     #Annotations that will be printed on the pdf
     single_annotations = create_annotations("single door", single_info)