Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -417,7 +417,7 @@ def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpo
 def main_run(img_pillow,pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ####pdf_fullpath here is the data and not the path 
     img_pillow = convert2pillow(pdf_fullpath)
     
-    num_classes = 10  # classes + background
+    num_classes = 12  # classes + background
     
     # Load the model with the specified number of classes
     model = get_model(num_classes)
@@ -438,7 +438,7 @@ def main_run(img_pillow,pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ###
     model.to(device)
     
     # START INFERENCE
-    doors_info = ev_model(img_pillow, model, device, 0.9)
+    doors_info = ev_model(img_pillow, model, device, 0.8)
     #width_pixels, lines, sanda, line_midpoint, doubleD_bbox, doubleD_width = get_door_info(doors_info)
     width_pixels, lines, sanda, line_midpoint, single_count, double_count, door_type = get_door_info(doors_info)
     real_width = pxl2meter(width_pixels, ratio)