YOLOv5 SKU-110预训练重量

Question

我想使用Yolov5来注释一个包含超市货架上产品的视频文件。有一个巨大的数据集(超过12 for ) SKU-110 K，在CVPR 2019年推出，包括成千上万的图片，许多超市货架培训Yolo。我在为SKU-110寻找训练过的YOLO重量。我在它的Github储存库和其他地方搜索，但什么也找不到。

• 有没有人知道YOLOv5曾接受过SKU-110 K的训练？
• 如果是的话，我在哪里可以找到它的重量档案？

Answer 1

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

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

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