Ultralytics官方给出的代码:
from ultralytics import YOLO
# Load a pre-trained YOLOv10n model
model = YOLO("yolov10n.pt")
# Perform object detection on an image
results = model("image.jpg")
# Display the results
results[0].show()
运行时出现:'dict' object has no attribute 'shape'.
我们提供的代码:
from ultralytics import YOLOv10 as YOLO import cv2 # Load a pre-trained YOLOv10n model model = YOLO("weights/last.pt") def predict_and_detect(chosen_model, img, classes=[], conf=0.8, rectangle_thickness=2, text_thickness=1): results = chosen_model.predict(img, classes, conf=conf) for result in results: for box in result.boxes: cv2.rectangle(img, (int(box.xyxy[0][0]), int(box.xyxy[0][1])), (int(box.xyxy[0][2]), int(box.xyxy[0][3])), (255, 0, 0), rectangle_thickness) cv2.put