yolov5识别红绿灯

以下是使用yolov5识别红绿灯的步骤：

1.首先，安装yolov5并下载交通信号灯检测的数据集。

2.然后，使用以下代码进行训练：

!python train.py --img 640 --batch 16 --epochs 30 --data traffic_lights.yaml --cfg models/yolov5s.yaml --weights yolov5s.pt --name yolov5s_traffic_lights

3.训练完成后，使用以下代码进行推理：

import torch
import cv2
from PIL import Image
import numpy as np

model = torch.hub.load('ultralytics/yolov5', 'custom', path_or_model='yolov5s_traffic_lights.pt')

img = Image.open('test.jpg')
results = model(img)

results.print()
results.show()

4.最后，你将得到一个包含检测结果的图像，其中包括红绿灯的位置和类别。