yifehuang97 commited on
Commit
288ef96
·
1 Parent(s): a77ab47

(feat) update neg processing

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -195,13 +195,13 @@ def count_objects(image, pos_caption, neg_caption, box_threshold, point_radius,
195
  fill=point_color
196
  )
197
 
198
- for point in neg_points:
199
- x = point[0] * img_w
200
- y = point[1] * img_h
201
- draw.ellipse(
202
- [x - point_radius, y - point_radius, x + point_radius, y + point_radius],
203
- fill="red"
204
- )
205
 
206
  count = len(points)
207
 
 
195
  fill=point_color
196
  )
197
 
198
+ # for point in neg_points:
199
+ # x = point[0] * img_w
200
+ # y = point[1] * img_h
201
+ # draw.ellipse(
202
+ # [x - point_radius, y - point_radius, x + point_radius, y + point_radius],
203
+ # fill="red"
204
+ # )
205
 
206
  count = len(points)
207