如果我想训练一个对象分离模型,它可以检测图片中的5个类,那么在一个大的数据集中(例如coco(80类对象))对这个模型进行预训练是很重要的,还是只需要5类coco来训练这个模型(假设这5个类别可以在coco中找到)?
发布于 2021-04-04 05:13:07
如果您想要检测的5个类已经在MS数据集中,则有两个可能的选项。
现在,如果您希望检测的类不在原始的MS-COCO数据集中,那么在早期的卷积层中使用预先训练的MS权重(在80个类上进行培训,即使它们与您的类无关),然后在您的数据集中训练网络的检测层和更深层,您会得到更好的效果。这是因为低层次的特征(如边缘,气泡等)。网络学到的大部分知识将是所有课程的共同之处,并将大大加快培训的速度。
https://stackoverflow.com/questions/66929506
复制相似问题