tensorrtx-master yolov5 python精简代码
官方源码tensorrtx-master\yolov5目录下提供很多cpp和py的案例,博主这里给大家提供一个yolov5_det_trt.py和yolov5_seg_trt.py的精简代码,代码中剔除了很多辅助性的功能函数。
主要功能就是读取一张本地图片,并进行目标预测,博主这里以车牌识别为主,提供工程源码
代码结构如下:

其中yolov5_det_trt.py和yolov5_seg_trt.py是tensorrtx-master\yolov5的源码,mydet.py和myseg.py是对应的精简代码。
运行结果如下:
