yolo算法检测框上表现中文，yolo表现中文

 
检测效果可以看我b站视频【yolo检测框表现中文-哔哩哔哩】
https://b23.tv/iXZRt6m

2. from ultralytics import YOLO
3. import numpy as np
4. import cv2
5. from ultralytics import YOLO
7. # Load a pretrained YOLOv8n model
8. model = YOLO("yolov8n.pt")#我加载的是官方权重
10. # Define path to video file
11. video_path = r"D:\daye_input.mp4"
12. cap = cv2.VideoCapture(video_path)
13. if not cap.isOpened():
14.     print("Cannot open camera")
15.     exit()
17. fps = cap.get(cv2.CAP_PROP_FPS)#输入视频帧率
18. print(f"输入视频帧率为：Frames per second: {fps}")
20. # 保存视频的一些设置
21. width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
22. height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
23. # width = int(1920)
24. # height = int(1080)
25. # fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 保存视频的编码格式
26. # output_video_path = 'daye.mp4'
27. # fourcc = cv2.VideoWriter_fourcc(*'mp4v')  # 保存视频的编码格式
28. # output_video_path = 'daye.mp4'
30. fourcc = cv2.VideoWriter_fourcc(*'XVID')  # 保存视频的编码格式
31. output_video_path = 'daye.avi'
34. out = cv2.VideoWriter(output_video_path, fourcc, fps, (width, height))
36. from PIL import Image, ImageDraw, ImageFont
38. # 设置字体
39. font = ImageFont.truetype(r'C:\Windows\Fonts\SimHei.ttf', 60)  # 替换为你的字体路径
40. # 循环遍历视频帧
41. while cap.isOpened():
43.     # start_time = time.time() #记录开始时间
45.     # 从视频读取一帧
46.     success, frame = cap.read()
48.     if success:
49.         # 在帧上运行YOLOv8追踪，持续追踪帧间的物体
50.         #     results = model(frame,  conf=0.3,iou=0.5,imgsz=(640,640))
51.             results = model.track(frame, persist=True, conf=0.3, iou=0.5, tracker="ultralytics/cfg/trackers/bytetrack.yaml",
52.                               imgsz=(1920,1080))
53.             img_pil = Image.fromarray(frame)
54.             draw = ImageDraw.Draw(img_pil)  # 创建Draw对象
56.             if results[0].boxes and results[0].boxes.id is not None:
57.                boxes = results[0].boxes.xyxy.cpu()
58.                clss = results[0].boxes.cls
59.                track_ids = results[0].boxes.id.int().cpu().tolist()
60.                for id, cls, boxxyxy in zip( track_ids, clss, boxes ):
61.                    class_id = cls.item()
62.                    x1, y1, x2, y2 = boxxyxy
63.                    tracker_id = id
64.                    if class_id == 0.0:
65.                        # class_id = 'DaYe'
67.                        draw.text((int(x1), int(y1) - 50), str(tracker_id)+"号大爷", font=font, fill=(0, 0,255))# 在图片上绘制中文
68.                        # cv2.rectangle(img_pil, (int(x1), int(y1)), (int(x2), int(y2)), (0, 69, 255), 2)  # 橙红
70.                        draw.rectangle([(int(x1), int(y1)), (int(x2), int(y2))], outline="blue", width=4)  # 红色边框，宽度5
72.                    # else:
73.                    #     draw.text((int(x1), int(y1) - 50),  "不像大爷", font=font,
74.                    #               fill=(255, 0, 0))  # 在图片上绘制中文
75.                    #     # cv2.rectangle(img_pil, (int(x1), int(y1)), (int(x2), int(y2)), (0, 69, 255), 2)  # 橙红
76.                    #
77.                    #     draw.rectangle([(int(x1), int(y1)), (int(x2), int(y2))], outline="red", width=4)  # 红色边框，宽度5
79.             # img = cv2.cvtColor(np.array(img_pil), cv2.COLOR_BGR2RGB)
80.             img = np.array(img_pil)
81.             # cv2.imshow('frame_pil', img)
82.             # img = cv2.resize( img, (1920, 1080))
83.             out.write(img)
85.             cv2.imshow('frame_pil', img)
86.             if cv2.waitKey(1) & 0xFF == ord('q'):
87.                 break
88.             # cv2.waitKey(0)
89.             # cv2.destroyAllWindows()
90.     else:
91.         break
92. cap.release()
93. out.release()
94. cv2.destroyAllWindows()

复制代码

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！qidao123.com:ToB企服之家，中国第一个企服评测及软件市场,开放入驻,技术点评得现金