模型訓練方法、電子設備和計算機可讀存儲介質(zhì)

基本信息

申請?zhí)?/td> CN202111614740.0 申請日 -
公開（公告）號 CN114005015A 公開（公告）日 2022-02-01
申請公布號 CN114005015A 申請公布日 2022-02-01
分類號 G06V10/774（2022.01）I;G06K9/62（2022.01）I 分類 計算；推算；計數(shù);
發(fā)明人 浦煜;何武;付賢強;朱海濤;戶磊 申請（專利權）人 合肥的盧深視科技有限公司
代理機構 北京智晨知識產(chǎn)權代理有限公司 代理人 張婧
地址 100083北京市海淀區(qū)學院路甲5號2幢平房B北3032室
法律狀態(tài) -

摘要

摘要 本申請實施例涉及視覺搜索技術領域，公開了一種模型訓練方法、電子設備和計算機可讀存儲介質(zhì)，該方法包括：獲取第一模型的訓練樣本；其中，訓練樣本標注用于表征訓練樣本的特征的特征類別的標簽；基于第一模型的網(wǎng)絡結構，構建第二模型；根據(jù)訓練樣本和第三模型，獲取第三模型對應的各特征類別的類別中心向量；其中，第一模型和第三模型為功能相同的模型；根據(jù)各特征類別的類別中心向量，確定第二模型的分類層權重；根據(jù)訓練樣本和標簽對第二模型進行迭代訓練，更新第二模型的除分類層權重之外的參數(shù)，訓練出的模型提取出的特征集可以直接與用舊模型提取出的特征庫進行對比，省時省力，降低了成本，極大地提升了模型工業(yè)部署的便利性。