Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -253,7 +253,7 @@ def adjustannotations(OutputPdfStage1):
   print("Annotations updated and saved ")
   return output_pdf_io.read()
 
-def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpoint, doubleD_bbox):
+def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpoint, doubleD_bbox, char_d_width):
     image_width, image_height = image.size
 
     # Create a new PDF document
@@ -284,6 +284,16 @@ def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpo
       annot = page.add_freetext_annot(rect, text, fontsize=10, fontname="helv", text_color=(1, 0, 0))
       annot.update()
 
+    for i in range(len(doubleD_bbox)):
+        #a7seb ana el midpoint beta3 el bbox 3ashan a7ot el width feha 
+        x,y = calculate_midpoint((doubleD_bbox[i][0], doubleD_bbox[i][1]), (doubleD_bbox[i][2],doubleD_bbox[i][3]))
+        p_midpoint = fitz.Point(x, y) * derotationMatrix
+        rect = fitz.Rect(p_midpoint.x, p_midpoint.y, p_midpoint.x + 200, p_midpoint.y + 50)
+        text = char_d_width[i]
+        annot = page.add_freetext_annot(rect, text, fontsize=10, fontname="helv", text_color=(1, 0, 0))
+        annot.update()
+
+        
     #Annotation for drawin lines as in the markups
     for i in range(len(lines)):
         l_points = [fitz.Point(*sanda[i][0])* derotationMatrix, fitz.Point(*lines[i][0])* derotationMatrix, fitz.Point(*lines[i][1])* derotationMatrix, fitz.Point(*sanda[i][1])* derotationMatrix]
@@ -388,14 +398,16 @@ def main_run(img_pillow,pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ###
     doors_info = ev_model(img_pillow, model, device, 0.9)
     width_pixels, lines, sanda, line_midpoint, doubleD_bbox, doubleD_width = get_door_info(doors_info)
     real_width = pxl2meter(width_pixels, ratio)
+    real_double = pxl2meter(doubleD_width, ratio)
     char_width = width_as_char(real_width)
+    char_d_width = width_as_char(real_double)
 
     pdf_document = add_annotations_to_pdf(img_pillow, pdf_fullpath, lines, sanda, char_width, line_midpoint, doubleD_bbox)
     modified_pdf_data = pdf_document.tobytes()
     OutputPdfStage2=adjustannotations(modified_pdf_data)
     #Dataframe for Doors count
-    #doors_count = {'Type': ['Single Doors', 'Double Doors'], 'Quantity': [len(sbox), len(dbox)]}
-    doors_count = {'Type': ['Single Doors'], 'Quantity': [len(real_width)]}
+    doors_count = {'Type': ['Single Doors', 'Double Doors'], 'Quantity': [len(real_width), len(real_double)]}
+    #doors_count = {'Type': ['Single Doors'], 'Quantity': [len(real_width)]}
     df_doors = pd.DataFrame(doors_count)
     
     doc2 =fitz.open('pdf',OutputPdfStage2)