Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -35,7 +35,7 @@ def yolov8_inference(
|
|
| 35 |
|
| 36 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 37 |
model = YOLO("https://huggingface.co/spaces/devisionx/Sixth_Demo/blob/main/bestt_weight.pt")
|
| 38 |
-
results = model(image,imgsz=640)[0]
|
| 39 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 40 |
detections = sv.Detections.from_yolov8(results)
|
| 41 |
annotated_image = annotatormask.annotate(scene=image, detections=detections)
|
|
@@ -61,7 +61,7 @@ title = "Tennis Court Demo"
|
|
| 61 |
'''
|
| 62 |
import os
|
| 63 |
examples = [
|
| 64 |
-
["t1.jpg", 0.
|
| 65 |
["t2.jpg", 0.25, 0.45],
|
| 66 |
["t3.jpg", 0.25, 0.45],
|
| 67 |
]
|
|
|
|
| 35 |
|
| 36 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 37 |
model = YOLO("https://huggingface.co/spaces/devisionx/Sixth_Demo/blob/main/bestt_weight.pt")
|
| 38 |
+
results = model(image,imgsz=640,conf=conf_threshold,iou=iou_threshold)[0]
|
| 39 |
image=image[:, :, ::-1].astype(np.uint8)
|
| 40 |
detections = sv.Detections.from_yolov8(results)
|
| 41 |
annotated_image = annotatormask.annotate(scene=image, detections=detections)
|
|
|
|
| 61 |
'''
|
| 62 |
import os
|
| 63 |
examples = [
|
| 64 |
+
["t1.jpg", 0.25, 0.45],
|
| 65 |
["t2.jpg", 0.25, 0.45],
|
| 66 |
["t3.jpg", 0.25, 0.45],
|
| 67 |
]
|