Segmentation of malaria-infected erythrocytes using U-Net incorporating Transformer and ResNet
10.3969/j.issn.1005-202X.2024.02.011
- VernacularTitle:采用融合ResNet和Transformer的U-Net进行疟疾感染红细胞分割
- Author:
Xiaoshuang LIU
1
;
Wei ZHANG
Author Information
1. 甘肃中医药大学信息工程学院,甘肃兰州 730000
- Keywords:
malaria;
U-Net;
Transformer;
semantic segmentation
- From:
Chinese Journal of Medical Physics
2024;41(2):191-197
- CountryChina
- Language:Chinese
-
Abstract:
A novel U-Net network model which integrates ResNet and Transformer is proposed to address the problem of poor malaria-in fected erythrocyte performance of the existing models.ResNet is used in the encoder to deepen the feature extraction network for extracting deeper features,and the ResNet output is inputted into Transformer module for the feature enhancement in the target area,and finally the decoder module is used to perform feature fusion and output the results.The experiment on the malaria microscopy image dataset shows that the proposed method outperforms U-Net in Dice similarity coefficient,mean intersection over union,and mean pixel accuracy,reaching 87.40%,76.85%,and 85.28%,respectively.The proposed method can improve the accuracy of malaria-infected erythrocyte segmentation and provide a more effective and accurate solution for malaria diagnosis.
