Establishment of a deep learning-based visual model for intelligent recognition of Oncomelania hupensis / 中国血吸虫病防治杂志

ABSTRACT

Objective To establish a deep learning-based visual model for intelligent recognition of Oncomelania hupensis, the intermediate host of Schistosoma japonicum, and evaluate the effects of different training strategies for O. hupensis image recognition. Methods A total of 2 614 datasets of O. hupensis snails and 4 similar snails were generated through field sampling and internet capture, and were divided into training sets and test sets. An intelligent recognition model was created based on deep learning, and was trained and tested. The precision, sensitivity, specificity, accuracy, F1 score and Youden index were calculated. In addition, the receiver operating characteristic (ROC) curve of the model for snail recognition was plotted to evaluate the effects of “new learning”, “transfer learning” and “transfer learning + data enhancement” training strategies on the accuracy of the model for snail recognition. Results Under the “transfer learning + data enhancement” strategy, the precision, sensitivity, specificity, accuracy, Youden index and F1 score of the model were 90.10%, 91.00%, 97.50%, 96.20%, 88.50% and 90.51% for snail recognition, which were all higher than those under both “new learning” and “transfer learning” strategies. There were significant differences in the sensitivity, specificity and accuracy of the model for snail recognition under “new learning”, “transfer learning” and “transfer learning + data enhancement” training strategies (all P values < 0.001). In addition, the area under the ROC curve of the model was highest (0.94) under the “transfer learning + dataenhancement” training strategy. Conclusions This is the first visual model for intelligent recognition of O. hupensis based on deep learning, which shows a high accuracy for snail image recognition. The “transfer learning + data enhancement” training strategy is helpful to improve the accuracy of the model for snail recognition.