Spaces:
Sleeping
Sleeping
Update doors_fasterrcnn.py
Browse files- doors_fasterrcnn.py +3 -3
doors_fasterrcnn.py
CHANGED
|
@@ -164,7 +164,7 @@ def get_door_info(doors_info):
|
|
| 164 |
lines.append((point_st,point_end))
|
| 165 |
line_midpoint.append(calculate_midpoint(point_st,point_end))
|
| 166 |
|
| 167 |
-
|
| 168 |
#sand_end = (int(xmax), int(ymax))
|
| 169 |
#sanda.append((sanda_st, sand_end))
|
| 170 |
#line_midpoint.append(calculate_midpoint(sanda_st,sand_end))
|
|
@@ -464,12 +464,12 @@ def main_run(img_pillow,pdf_fullpath, weights_path, weights_path2, pdf_name,pdfp
|
|
| 464 |
|
| 465 |
width_pixels, lines, line_midpoint, single_count, door_type = get_door_info(doors_info)
|
| 466 |
|
| 467 |
-
width_pixels, lines, line_midpoint, double_count, door_type = get_door_info_double(doors_info_double, width_pixels, lines,
|
| 468 |
|
| 469 |
real_width = pxl2meter(width_pixels, ratio)
|
| 470 |
char_width = width_as_char(real_width)
|
| 471 |
|
| 472 |
-
pdf_document = add_annotations_to_pdf(img_pillow, pdf_fullpath, lines,
|
| 473 |
modified_pdf_data = pdf_document.tobytes()
|
| 474 |
OutputPdfStage2=adjustannotations(modified_pdf_data)
|
| 475 |
|
|
|
|
| 164 |
lines.append((point_st,point_end))
|
| 165 |
line_midpoint.append(calculate_midpoint(point_st,point_end))
|
| 166 |
|
| 167 |
+
#sanda_st = (int(xmax), int(ymin))
|
| 168 |
#sand_end = (int(xmax), int(ymax))
|
| 169 |
#sanda.append((sanda_st, sand_end))
|
| 170 |
#line_midpoint.append(calculate_midpoint(sanda_st,sand_end))
|
|
|
|
| 464 |
|
| 465 |
width_pixels, lines, line_midpoint, single_count, door_type = get_door_info(doors_info)
|
| 466 |
|
| 467 |
+
width_pixels, lines, line_midpoint, double_count, door_type = get_door_info_double(doors_info_double, width_pixels, lines, line_midpoint, door_type)
|
| 468 |
|
| 469 |
real_width = pxl2meter(width_pixels, ratio)
|
| 470 |
char_width = width_as_char(real_width)
|
| 471 |
|
| 472 |
+
pdf_document = add_annotations_to_pdf(img_pillow, pdf_fullpath, lines, char_width, line_midpoint, door_type)
|
| 473 |
modified_pdf_data = pdf_document.tobytes()
|
| 474 |
OutputPdfStage2=adjustannotations(modified_pdf_data)
|
| 475 |
|