YOLOv5 SKU-110 预训练权重

Question

我想用Yolov5标注一个包含超市货架上商品的视频文件。有一个巨大的数据集（超过 12GiB）SKU-110K，在 CVPR 2019 中引入，包括许多超市货架的数千张图像，用于训练 Yolo。我正在为 SKU-110 寻找预训练的 YOLO 权重。我在其 Github 存储库 和其他地方进行了搜索，但什么也找不到。

• 有谁知道曾经为 SKU-110K 训练过 YOLOv5 吗？
• 如果是， 我在哪里可以找到它的权重文件？

Answer 1

没有官方的 SKU-110K Yolov5 权重（也没有我所见的任何私人训练和开源）但是你可以通过一个简单的命令在这个数据集上训练你自己的 Yolov5 模型：

python train.py --data SKU-110K.yaml

这将自动下载、准备数据并开始训练