这里是一个简单的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

load a pre-trained model for

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐