wzzanthony7 commited on
Commit
424902e
·
verified ·
1 Parent(s): ec3307f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -0
app.py CHANGED
@@ -195,6 +195,10 @@ def generate_textual_description(box_info):
195
  if getOverlap(fra_box, zero_box) >= 0.5:
196
  kept_zero = False
197
  break
 
 
 
 
198
  if kept_zero:
199
  kept_zero_boxes.append(zero_box)
200
  kept_one_boxes = []
@@ -204,6 +208,10 @@ def generate_textual_description(box_info):
204
  if getOverlap(fra_box, one_box) >= 0.5:
205
  kept_one = False
206
  break
 
 
 
 
207
  if kept_one:
208
  kept_one_boxes.append(one_box)
209
  kept_zero_boxes.sort(key = lambda x: x[0])
 
195
  if getOverlap(fra_box, zero_box) >= 0.5:
196
  kept_zero = False
197
  break
198
+ for tick_box in class_summary['tick']:
199
+ if getOverlap(tick_box, zero_box) >= 0.5:
200
+ kept_zero = False
201
+ break
202
  if kept_zero:
203
  kept_zero_boxes.append(zero_box)
204
  kept_one_boxes = []
 
208
  if getOverlap(fra_box, one_box) >= 0.5:
209
  kept_one = False
210
  break
211
+ for tick_box in class_summary['tick']:
212
+ if getOverlap(tick_box, one_box) >= 0.5:
213
+ kept_one = False
214
+ break
215
  if kept_one:
216
  kept_one_boxes.append(one_box)
217
  kept_zero_boxes.sort(key = lambda x: x[0])