Update doors_fasterrcnn.py

doors_fasterrcnn.py

@@ -125,6 +125,7 @@ def get_door_info(doors_info):
       #for calculation
       width = distance((xmin,ymax), (xmax,ymax))
       width_pixels.append(width)
+      door_type.append(0)
       singles +=1
 
     #horz_upper
@@ -142,6 +143,7 @@ def get_door_info(doors_info):
       width = distance((xmin,ymin), (xmax,ymin))
       width_pixels.append(width)
       singles +=1
+      door_type.append(0)
 
     #vert_right
     if door_inf[1] == 4:
@@ -158,6 +160,7 @@ def get_door_info(doors_info):
       width = distance((xmax,ymin), (xmax,ymax))
       width_pixels.append(width)
       singles +=1
+      door_type.append(0)
     #vert_left
     if door_inf[1] == 5:
       #for drawing
@@ -173,6 +176,7 @@ def get_door_info(doors_info):
       width = distance((xmin,ymin), (xmin,ymax))
       width_pixels.append(width)
       singles +=1
+      door_type.append(0)
 
   return width_pixels, lines, sanda, line_midpoint, singles, door_type