一种基于自训练和先验引导的图像分割方法和装置

AITNT-国内领先的一站式人工智能新闻资讯网站
# 热门搜索 #
一种基于自训练和先验引导的图像分割方法和装置
申请号:CN202411074124
申请日期:2024-08-07
公开号:CN118587443B
公开日期:2024-12-03
类型:发明专利
摘要
本发明公开了一种基于自训练和先验引导的图像分割方法和装置,该方法包括:对原始图像数据集进行标注,生成对应的真实掩码;使用含有真实掩码的图像对伪掩码生成网络进行训练,训练好的伪掩码生成网络用于根据不含掩码的图像生成伪掩码;从掩码图像中提取边界,生成边界图像;所述掩码图像包括真实掩码图像和伪掩码图像;基于原始图像及其对应的掩码图像和边界图像拼接成三通道图像;所述三通道图像用于训练图像分割模型,训练好的图像分割模型用于对待求图像进行分割。在计算资源和数据有限的情况下,本发明方法可以最大限度的减少训练参数量的同时发挥现有大模型的优势。本发明方法还可以加强分割模型的先验知识,进一步提升图像分割的结果。
技术关键词
图像分割模型 原始图像数据 图像编码器 图像分割方法 解码器 canny边缘检测 图像拼接 教师 学生 三通道 低噪声 图像分割装置 网络 逻辑 索引