Marthee commited on
Commit
b6cf08c
·
verified ·
1 Parent(s): 4a6dcbd

Update doors_fasterrcnn.py

Browse files
Files changed (1) hide show
  1. doors_fasterrcnn.py +4 -0
doors_fasterrcnn.py CHANGED
@@ -125,6 +125,7 @@ def get_door_info(doors_info):
125
  #for calculation
126
  width = distance((xmin,ymax), (xmax,ymax))
127
  width_pixels.append(width)
 
128
 
129
  #horz_upper
130
  if door_inf[1] == 3:
@@ -140,6 +141,7 @@ def get_door_info(doors_info):
140
  #for calculation
141
  width = distance((xmin,ymin), (xmax,ymin))
142
  width_pixels.append(width)
 
143
 
144
  #vert_right
145
  if door_inf[1] == 4:
@@ -155,6 +157,7 @@ def get_door_info(doors_info):
155
  #for calculation
156
  width = distance((xmax,ymin), (xmax,ymax))
157
  width_pixels.append(width)
 
158
  #vert_left
159
  if door_inf[1] == 5:
160
  #for drawing
@@ -169,6 +172,7 @@ def get_door_info(doors_info):
169
  #for calculation
170
  width = distance((xmin,ymin), (xmin,ymax))
171
  width_pixels.append(width)
 
172
 
173
  return width_pixels, lines, sanda, line_midpoint, singles, door_type
174
 
 
125
  #for calculation
126
  width = distance((xmin,ymax), (xmax,ymax))
127
  width_pixels.append(width)
128
+ singles +=1
129
 
130
  #horz_upper
131
  if door_inf[1] == 3:
 
141
  #for calculation
142
  width = distance((xmin,ymin), (xmax,ymin))
143
  width_pixels.append(width)
144
+ singles +=1
145
 
146
  #vert_right
147
  if door_inf[1] == 4:
 
157
  #for calculation
158
  width = distance((xmax,ymin), (xmax,ymax))
159
  width_pixels.append(width)
160
+ singles +=1
161
  #vert_left
162
  if door_inf[1] == 5:
163
  #for drawing
 
172
  #for calculation
173
  width = distance((xmin,ymin), (xmin,ymax))
174
  width_pixels.append(width)
175
+ singles +=1
176
 
177
  return width_pixels, lines, sanda, line_midpoint, singles, door_type
178