一种基于RS loss的目标检测模型的训练方法及装置

摘要

本发明公开了一种基于RS loss的目标检测模型的训练方法及装置，该方法包括获取多个目标检测训练数据集，对多个目标检测训练数据集中的训练数据进行预处理，得到预处理数据，根据预处理数据，训练基于RS loss的Faster‑RCNN模型，得到目标检测模型，在目标检测模型基础上增加属性分支，并依据VG数据进行微调。通过在Faster‑RCNN基础上，将原有通过指定多任务超参数方式替换为基于RS loss，简化原有检测模型的复杂性，并避免因次优超参数而降低模型性能，提升了模型视觉表征能力。