摘要:構(gòu)建大規(guī)模茶芽目標檢測數(shù)據(jù)集是一項耗時且繁瑣的任務(wù),,為了降低數(shù)據(jù)集構(gòu)建成本,探索少量標注樣本的算法尤為必要,。本文提出了YSVD-Tea (YOLO singular value decomposition for tea bud detection)算法,,通過將預(yù)訓(xùn)練模型中的基礎(chǔ)卷積替換為3個連續(xù)的矩陣結(jié)構(gòu),實現(xiàn)了對YOLOX算法結(jié)構(gòu)的重構(gòu),。通過維度變化和奇異值分解操作,,將預(yù)訓(xùn)練權(quán)重轉(zhuǎn)換為與重構(gòu)算法結(jié)構(gòu)相對應(yīng)的權(quán)重,從而將需要進行遷移學(xué)習(xí)的權(quán)重和需要保留的權(quán)重分離開,,實現(xiàn)保留預(yù)訓(xùn)練模型先驗信息的目的,。在3種不同數(shù)量的數(shù)據(jù)集上分別進行了訓(xùn)練和驗證。在最小數(shù)量的1/3數(shù)據(jù)集上,,YSVD-Tea算法相較于改進前的YOLOX算法,,mAP提高20.3個百分點。對比測試集與訓(xùn)練集的性能指標,,YSVD-Tea算法在測試集與訓(xùn)練集的mAP差距僅為21.9%,,明顯小于YOLOX的40.6%和Faster R-CNN的55.4%。在數(shù)量最大的數(shù)據(jù)集上,,YOLOX算法精確率,、召回率,、F1值、mAP分別為86.4%,、87.0%,、86.7%和88.3%,相較于對比算法均最高,。YSVD-Tea在保證良好性能的同時,,能夠更好地適應(yīng)少量標注樣本的茶芽目標檢測任務(wù)。