基于元学习的小样本轮胎印迹语义分割算法

FEW-SHOT SEMANTIC SEGMENTATION FOR SAMPLE TIRE MARKS BASED ON META-LEARNING

  • 摘要: 针对事件现场调查中人工比对碾压轮胎印迹效率低的问题,提出一种基于元学习的小样本轮胎印迹语义分割算法。利用预测的特征提取器将轮胎印迹图片映射到深度特征空间,并基于Vision Transformer网络构建一种自适应的目标分类器,借助理论学习的方法细粒度的微调整个模型。结果表明,该模型在1-shot和5-shot分割任务上可以实现较好的分割效果,相比当前主流的分割模型,具有更好的分割性能。

     

    Abstract: Aimed at the low efficiency of manual comparison of tire tracks on roller tracks in the scene investigation, a few-shot semantic segmentation algorithm for small sample tire tracks based on meta-learning is proposed. A pre-trained feature extractor was used to map support images to the depth feature space, and the adaptive target classifier was constructed based on vision transformer (ViT). The whole model was optimized using the reinforcement learning in the fine-tuned manner. The results show that the proposed model can achieve the better segmentation performance in 1-shot and 5-shot segmentation tasks, and has better performance than the current mainstream segmentation models.

     

/

返回文章
返回