Marthee commited on
Commit
50d34b9
·
verified ·
1 Parent(s): bd40344

Update doors_fasterrcnn.py

Browse files
Files changed (1) hide show
  1. doors_fasterrcnn.py +2 -2
doors_fasterrcnn.py CHANGED
@@ -423,7 +423,7 @@ def add_annotations_to_pdf(image, pdf_name, lines, sanda, char_width, line_midpo
423
 
424
 
425
 
426
- def main_run(img_pillow,pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ####pdf_fullpath here is the data and not the path
427
  img_pillow = convert2pillow(pdf_fullpath)
428
 
429
  # For Single Doors
@@ -460,7 +460,7 @@ def main_run(img_pillow,pdf_fullpath, weights_path, pdf_name,pdfpath,ratio): ###
460
  model2.to(device)
461
 
462
  # START INFERENCE
463
- doors_info = ev_model(img_pillow, model, device, 0.8)
464
  doors_info_double = ev_model(img_pillow, model2, device, 0.8)
465
 
466
 
 
423
 
424
 
425
 
426
+ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfpath,ratio): ####pdf_fullpath here is the data and not the path
427
  img_pillow = convert2pillow(pdf_fullpath)
428
 
429
  # For Single Doors
 
460
  model2.to(device)
461
 
462
  # START INFERENCE
463
+ doors_info = ev_model(img_pillow, model, device, 0.9)
464
  doors_info_double = ev_model(img_pillow, model2, device, 0.8)
465
 
466