25 lines
857 B
Python
25 lines
857 B
Python
|
from ultralytics import YOLO
|
|||
|
|
import cv2
|
|||
|
|
import torch
|
|||
|
|
|
|||
|
|
# 加载模型
|
|||
|
|
model = YOLO('best.pt')
|
|||
|
|
|
|||
|
|
# 读取一张真实图像
|
|||
|
|
img_path = '/media/hx/04e879fa-d697-4b02-ac7e-a4148876ebb0/dataset/point2/train/1.jpg' # 替换成您的图像路径
|
|||
|
|
image = cv2.imread(img_path)
|
|||
|
|
|
|||
|
|
# 将图像转换成RGB格式，并调整大小
|
|||
|
|
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
|
|||
|
|
image_resized = cv2.resize(image_rgb, (640, 640))
|
|||
|
|
|
|||
|
|
# 进行推理
|
|||
|
|
results = model(image_resized)
|
|||
|
|
|
|||
|
|
# 打印关键点数据形状和样本
|
|||
|
|
if len(results) > 0 and hasattr(results[0], 'keypoints') and results[0].keypoints is not None:
|
|||
|
|
print("Keypoints data shape:", results[0].keypoints.data.shape)
|
|||
|
|
if results[0].keypoints.data.shape[0] > 0:
|
|||
|
|
print("Keypoints data sample:", results[0].keypoints.data[0, :12])
|
|||
|
|
else:
|
|||
|
|
print("No keypoints detected or invalid keypoints data.")
|