用python写一个yolov5的游戏角色识别程序

YOLOv5 是一种实时物体检测算法，可以用来识别图像中的游戏角色。在使用 YOLOv5 识别游戏角色时，首先需要获取训练好的模型权重文件，然后使用 Python 编写代码进行识别。下面是一个简单的示例代码，使用 YOLOv5 模型和 PyTorch 库进行游戏角色识别：

import cv2
import torch
from yolov5.models import YOLOv5

# 加载预训练模型权重
model = YOLOv5(weights='weights/best.pt')
model.eval()

# 读取图像
img = cv2.imread('image.jp