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