Delete temp.py

This commit is contained in:
Bhavay Malhotra authored and GitHub committed 2024-08-09 23:20:05 +05:30
1 parent 78689590e6
commit 2b73aea438
1 file changed
-40
-40
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)