From 2b73aea43817490138ffee3e17bf96646feb8e8a Mon Sep 17 00:00:00 2001 From: Bhavay Malhotra <56443877+Bhavay-2001@users.noreply.github.com> Date: Fri, 9 Aug 2024 23:20:05 +0530 Subject: [PATCH] Delete temp.py --- temp.py | 40 ---------------------------------------- 1 file changed, 40 deletions(-) delete mode 100644 temp.py diff --git a/temp.py b/temp.py deleted file mode 100644 index 76c0d9cd..00000000 --- a/temp.py +++ /dev/null @@ -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)