Marthee commited on
Commit
336ddd1
·
verified ·
1 Parent(s): c2ceb86

Update doors_fasterrcnn.py

Browse files
Files changed (1) hide show
  1. 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
- #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,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, sanda, 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, sanda, char_width, line_midpoint, door_type)
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