Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -386,7 +386,7 @@ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfp
     img_pillow = convert2pillow(pdf_fullpath)
 
     # For Single Doors
-    num_classes = 12  # classes + background
+    num_classes = 10 # classes + background
     # Load the model with the specified number of classes
     model = get_model(num_classes)
     # Load the saved model's state dictionary with map_location to handle CPU
@@ -403,7 +403,7 @@ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfp
     
 
     # For Double Doors
-    num_classes2 = 11  # classes + background
+    num_classes2 = 12  # classes + background
     # Load the model with the specified number of classes
     model2 = get_model(num_classes2)
     # Load the saved model's state dictionary with map_location to handle CPU