使用场景:自己定义的关键点比如说做了耳蜗检测有13个关键点(就是原来的pose点数了)可以单独修改每个关键点的颜色以及Label的颜色。
代码:
import cv2
from ultralytics import YOLO
import torch
import numpy as np
class PoseDetector:
def __init__(self, model_path):
self.device = torch.device
使用场景:自己定义的关键点比如说做了耳蜗检测有13个关键点(就是原来的pose点数了)可以单独修改每个关键点的颜色以及Label的颜色。
代码:
import cv2
from ultralytics import YOLO
import torch
import numpy as np
class PoseDetector:
def __init__(self, model_path):
self.device = torch.device