Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -100,6 +100,7 @@ def get_door_info(doors_info):
   lines = []
   sanda = []
   line_midpoint = []
+  single_door = 0
   for door_inf in doors_info:
     xmin, ymin, xmax, ymax = door_inf[0]
     #horz_bottom
@@ -117,6 +118,7 @@ def get_door_info(doors_info):
       #for calculation
       width = distance((xmin,ymax), (xmax,ymax))
       width_pixels.append(width)
+      #single_door +=1
 
     #horz_upper
     if door_inf[1] == 3:
@@ -245,7 +247,11 @@ def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ####pdf_fullpa
     char_width = width_as_char(real_width)
 
     pdf_document = add_annotations_to_pdf(img_pillow, plan, lines, sanda, char_width, line_midpoint)
-    
+
+    #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)]}
+    df_doors = pd.DataFrame(doors_count)
 
     page=pdf_document[0]
     pix = page.get_pixmap()  # render page to an image
@@ -262,7 +268,7 @@ def main_run(pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ####pdf_fullpa
 
 
     # modify this return
-    return annotatedimg, pdf_document , spreadsheet_url, list1 
+    return annotatedimg, pdf_document , spreadsheet_url, list1, df_doors 
 
 # model_path =  '/content/drive/MyDrive/combined.pth'
 # #pdf_name = data