PIG FACE IDENTIFICATION BASED ON IMPROVED ALEXNET MODEL
基于改进AlexNet模型的生猪脸部识别
DOI : https://doi.org/10.35633/inmateh-61-11
Authors
(*) Corresponding authors:
Abstract
Individual pig identification technology is the precondition of precise breeding. Taking pig face as the study point, this article puts forward a pig face identification method based on improved AlexNet model and explores the influence of training batch size on the performance of the model. Spatial attention module (SAM) is introduced in AlexNet model to compare the performance of the AlexNet model and the improved model on the training set and the validation set. The study shows that the improved AlexNet model can achieve higher precision rate under different training batch sizes and has higher convergence rate and robustness, with an identification precision rate reaching 98.11%, and a recall rate and f1 value reaching 98.03% and 98.05%. When the training batch sizes are 16, 32, and 64 respectively, the test time of the model, which represents its operating efficiency, improves by 1.99%, 2.36% and 10.31%, respectively, showing better performance in pig face identification. The test results show that different batch sizes have a certain influence on the prediction results of the model, while no fixed relationship.
Abstract in Chinese