Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -199,17 +199,22 @@ def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ####pdf_fullpa
     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, ratio)
-    double_info = mid_points_bbox(dbox, ratio)
+    single_midpoint = mid_points_bbox(sbox)
+    double_midpoint = mid_points_bbox(dbox)
+
+    #Kind Annotations
+    single_annotations = create_annotations("single door", single_midpoint)
+    double_annotations = create_annotations("double door", double_midpoint)
+    #Lines Annotations
+    line_single, width_signle = width_annotations(sbox, 1)
+    line_double, width_double = width_annotations(dbox, 1)
+    #Width Annotations
+    width_single_ann = create_width_annotations(width_signle)
+    width_double_ann = create_width_annotations(width_double)
+
+    # add_annotations_to_pdf(new_image, pdf_name, single_annotations, double_annotations)
+    add_annotations_to_pdf(img_pillow, pdf_name, single_annotations, double_annotations,width_single_ann,width_double_ann,line_single,line_double)
 
-    #Annotations that will be printed on the pdf
-    single_annotations = create_annotations("single door", single_info)
-    double_annotations = create_annotations("double door", double_info)
-    
-        # add_annotations_to_pdf(new_image, pdf_name, single_annotations, double_annotations)
-    pdf_document=add_annotations_to_pdf(img_pillow, pdf_name, single_annotations, double_annotations)
-    
     page=pdf_document[0]
     pix = page.get_pixmap()  # render page to an image
     pl=Image.frombytes('RGB', [pix.width,pix.height],pix.samples)