约洛夫5
来自原始代码。 我修改了yololayer并集成了批处理的NMSPlugin。 提供了yolov5s.wts可以进行快速演示。 如何生成.wts可以参考 。
如何运行,以yolov5s为例
建立并运行
mkdir build
cd build
cmake ..
make
sudo ./yolov5 -s // serialize model to plan file i.e. 'yolov5s.engine'
sudo ./yolov5 -d ../samples // deserialize plan file and run inference, the images in samples will be processed.
检查生成的图像,如下所示。 _zidane.jpg和_bus.jpg
运行Python示例,请安装Python ten