这里是一个简单的YOLOv5代码,可以用来识别红绿灯图像:
import torch import torchvision
from torchvision.models.detection.faster_rcnn import FastRCNNPredictor from torchvision.models.detection.mask_rcnn import MaskRCNNPredictor
from torchvision.models.detection.rpn import AnchorGenerator