Spaces:
Sleeping
Sleeping
Update doors_fasterrcnn.py
Browse files- 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.
|
| 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 |
|