search

wudashuo / yolov5

YOLOv5 汉化版，保持官方同步更新
GNU General Public License v3.0
445 stars 100 forks source link

分类种类太多该怎么调优？ #7

Open chenmonster opened 3 years ago

chenmonster commented 3 years ago

❔Question

运行的时，有500多种分类，批次大小32，然后就报GPU内存不足。后面就把种类调到85，批次大小调成16就没报错了，请问这个怎么调参呢？

Additional context

wudashuo commented 3 years ago

和类别数关系不大，报out of memory的话：

  1. 调小batch size
  2. 用更小的模型(yolov5s)
  3. 用更小的输入图片尺寸
  4. 尝试调整workers，也能降低GPU显存使用量

另外，为什么你的数据集会有500种类别？用的自己的数据集？你这个明显不合理。 根据官方建议，你的数据集总数量应大于类别数 1500，目标总数大于10000 类别数