Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -194,9 +194,15 @@ def main_run(pdf_fullpath, weights_path, pdf_name):
     # START INFERENCE
     sbox, dbox = ev_model(new_image, 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)
-    
+
+    #Annotations that will be printed on the pdf
     single_annotations = create_annotations("single door", single_info)
     double_annotations = create_annotations("double door", double_info)