重庆工商大学学报（自然科学版）2024，Vol.41Issue(5)：65-71,7.DOI:10.16055/j.issn.1672-058X.2024.0005.008
基于U-net和胶囊网络的图像语义分割结构研究
Research on Image Semantic Segmentation Structure Based on U-net and Capsule Network
摘要
Abstract
Objective Mosaic disease is a common symptom in apples.Especially under the condition of large temperature differences between day and night,the onset of mosaic disease is rapid,which can lead to an increase in the rate of defoliation,resulting in a large reduction in apple production and huge economic losses.The number of mosaic disease spots is too high and the scale of the spots varies,resulting in problems such as low accuracy of disease identification.Based on this,a method that introduces transfer learning and capsule networks was proposed to improve the disease identification rate.Methods Firstly,this method expanded and enhanced the obtained mosaic disease dataset,and used the Labelme tool to annotate images,marking the lesion area and leaf area respectively.Secondly,the weight of the trained VGG16 model was transferred to the coding part of U-net through the transfer learning technology,and the capsule network was introduced,so that the whole network had a stronger feature extraction ability.Then,the VGG16 model and capsule network were trained.Finally,the trained network model was semantically segmented and test results were output.Results The experimental results showed that the accuracy rate of the original dataset was 87.51％,and the accuracy rate after the introduction of transfer learning was improved to 91.78％,an improvement of 4.88％;the accuracy of introducing the capsule network was improved to 90.04％,an increase of 2.89％;after the introduction of transfer learning and capsule network,the accuracy rate was improved to 93.42％,an improvement of 6.75％.In addition,the training time of each round of the model was also improved by 2 s after the introduction of transfer learning.Conclusion According to the experimental results,it can be proved that the proposed model,after the introduction of transfer learning and capsule network,has a certain improvement in identification accuracy compared with the traditional model.Furthermore,this method also reduces the model training time in each round and the overall segmentation performance is better.关键词
病害识别/花叶病/病斑/VGG16/U-net/胶囊网络Key words
disease identification/mosaic disease/disease spots/VGG16/U-net/capsule network分类
信息技术与安全科学引用本文复制引用
刘向举,赵慧勐,方贤进..基于U-net和胶囊网络的图像语义分割结构研究[J].重庆工商大学学报（自然科学版）,2024,41(5):65-71,7.基金项目
国家自然科学基金项目(61572034) （61572034）
安徽省科技重大专项(18030901025). （18030901025）
