在yolo网站上,可可数据集上已预先训练了权重,权重由80个对象/类组成(我还认为由20个类组成的voc pascal数据集的权重)。 是否有(一些)已经在具有更多类的数据集中训练过的权重?我当时在想YOLO可以检测到数千个对象,而不仅仅是80个。还是我想念什么?
答案 0 :(得分:1)
您不太可能找到在合理的条件下训练超过或少于80个班级的重量。您可以从GitHub恢复完整的仓库pjreddie/darknet,然后更改配置文件,以根据所需数量的类训练模型。
完整说明在存储库的README.md文件中等待。