Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -167,7 +167,7 @@ async def predict_single_dog(image):
|
|
167 |
return top1_prob, topk_breeds, topk_probs_percent
|
168 |
|
169 |
|
170 |
-
async def detect_multiple_dogs(image, conf_threshold=0.
|
171 |
results = model_yolo(image, conf=conf_threshold, iou=iou_threshold)[0]
|
172 |
dogs = []
|
173 |
boxes = []
|
@@ -288,7 +288,9 @@ async def predict(image):
|
|
288 |
top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
|
289 |
color = color_list[i % len(color_list)]
|
290 |
draw.rectangle(box, outline=color, width=3)
|
291 |
-
draw.text((box[0], box[1]), f"Dog {i+1}", fill=color, font=font)
|
|
|
|
|
292 |
|
293 |
combined_confidence = detection_confidence * top1_prob
|
294 |
|
|
|
167 |
return top1_prob, topk_breeds, topk_probs_percent
|
168 |
|
169 |
|
170 |
+
async def detect_multiple_dogs(image, conf_threshold=0.4, iou_threshold=0.55):
|
171 |
results = model_yolo(image, conf=conf_threshold, iou=iou_threshold)[0]
|
172 |
dogs = []
|
173 |
boxes = []
|
|
|
288 |
top1_prob, topk_breeds, topk_probs_percent = await predict_single_dog(cropped_image)
|
289 |
color = color_list[i % len(color_list)]
|
290 |
draw.rectangle(box, outline=color, width=3)
|
291 |
+
#draw.text((box[0], box[1]), f"Dog {i+1}", fill=color, font=font)
|
292 |
+
draw.text((box[0] + 5, box[1] + 5), f"Dog {i+1}", fill=color, font=font)
|
293 |
+
|
294 |
|
295 |
combined_confidence = detection_confidence * top1_prob
|
296 |
|