xiaoxue_admin
/
supervision
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Delete temp.py

This commit is contained in:
Bhavay Malhotra 2024-08-09 23:20:05 +05:30 committed by GitHub
parent 78689590e6
commit 2b73aea438
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 40 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
temp.py
View File

@ -1,40 +0,0 @@
import cv2
from ultralytics import YOLO
import supervision as sv
image = cv2.imread("basketball.jpg")
model = YOLO("yolov8s-pose.pt")
result = model(image)[0]
key_points = sv.KeyPoints.from_ultralytics(result)
COLORS = [
"#FF6347",
"#FF6347",
"#FF6347",
"#FF6347",
"#FF6347",
"#FF1493",
"#00FF00",
"#FF1493",
"#00FF00",
"#FF1493",
"#00FF00",
"#FFD700",
"#00BFFF",
"#FFD700",
"#00BFFF",
"#FFD700",
"#00BFFF",
]
COLORS = [sv.Color.from_hex(color_hex=c) for c in COLORS]
vertex_label_annotator = sv.VertexLabelAnnotator(
color=COLORS, text_color=sv.ColorPalette.ROBOFLOW, border_radius=5
)
annotated_frame = vertex_label_annotator.annotate(
scene=image.copy(), key_points=key_points
)
sv.plot_image(annotated_frame)