Commit
·
e3fe9bf
1
Parent(s):
56ce5ee
Print acc
Browse files
app.py
CHANGED
|
@@ -331,12 +331,12 @@ async def detect_animals(file: UploadFile = File(...)):
|
|
| 331 |
results = model_animal(image)[0]
|
| 332 |
for box in results.boxes:
|
| 333 |
conf = box.conf[0].item()
|
| 334 |
-
if conf >= 0.
|
| 335 |
cls_id = int(box.cls[0].item())
|
| 336 |
label = model_animal.names[cls_id].lower()
|
| 337 |
if label in ["dog", "cat", "cow", "horse", "elephant", "bear", "zebra", "giraffe", "bird"]:
|
| 338 |
x1, y1, x2, y2 = map(int, box.xyxy[0].tolist())
|
| 339 |
-
detections.append(((x1, y1, x2, y2), "Animal Alert"))
|
| 340 |
except Exception as e:
|
| 341 |
print("[YOLOv8 Error]", e)
|
| 342 |
|
|
|
|
| 331 |
results = model_animal(image)[0]
|
| 332 |
for box in results.boxes:
|
| 333 |
conf = box.conf[0].item()
|
| 334 |
+
if conf >= 0.70:
|
| 335 |
cls_id = int(box.cls[0].item())
|
| 336 |
label = model_animal.names[cls_id].lower()
|
| 337 |
if label in ["dog", "cat", "cow", "horse", "elephant", "bear", "zebra", "giraffe", "bird"]:
|
| 338 |
x1, y1, x2, y2 = map(int, box.xyxy[0].tolist())
|
| 339 |
+
detections.append(((x1, y1, x2, y2), f"Animal Alert {conf}"))
|
| 340 |
except Exception as e:
|
| 341 |
print("[YOLOv8 Error]", e)
|
| 342 |
|