YoloV5的FPS太少了，怎么解决？答案

【问题标题】：YoloV5 way too less FPS, how can I fix it?YoloV5的FPS太少了，怎么解决？
【发布时间】：2021-12-30 00:10:13
【问题描述】：

你能帮我用 Yolov5s 提高我的 FPS 吗？

我在游戏中使用 yolov5s 进行实时检测，但我的 FPS 确实非常低（大约 30-40），有时只有 0。我将它与我的自定义数据集一起使用：

model = torch.hub.load('ultralytics/yolov5', 'custom', path=r'C:\Users\stefa\Downloads\best2.pt') # local repo

并用 mss 截屏：

with mss.mss() as sct:
    # Part of the screen to capture
    monitor = {"top": 40, "left": 0, "width": 800, "height": 640}

我希望有人可以帮助我提高我的 FPS。

或者我应该使用Yolov5n，如果是，我该如何使用/下载它？

谢谢！

【问题讨论】：

【解决方案1】：

有几种方法可以提高您的 FPS 速率：

越简单越好。尝试使用较小的模型，例如 YOLOv5s 或 YOLOv5n
使用 GPU 进行推理怎么样？添加这一行：--GPU。示例：$ python detect.py --source 0 --gpu #--source 0 = 网络摄像头，确保更改它。
将您的视野缩小到武器附近的一个小边界框（尝试使用 480x480）。也许您需要调整训练集的大小以满足此要求。
检查这个：https://www.youtube.com/watch?v=Yqwz4QGDNh0&ab_channel=SometimesRobots

祝你好运！

【讨论】：

嘿，我正在使用 CUDA，但是如何使用更小的模型，例如 yolov5n？你知道一个教程或我怎么能做到这一点？感谢您的帮助！
嘿@Dani，你可以在训练前改变模型如下： !python train.py --img 480 --batch 4 --epochs 100 --datadefects.yaml --cfg models/ yolov5s.yaml --weights yolov5s.pt --name your_trained_model --cache
批量越大还是越小越好？
这取决于您的 GPU 内存。批次越大，训练越快。从 14 开始并减少它。 PD。不要忘记重命名和更改您的 YAML 文件。
好的，谢谢。更多 FPS yolov5 或 yolov4 哪个更好？以及如何使用 yolov5n（纳米模型）？