Human Stools Classification for Gastrointestinal Health Based on an Improved ResNet18 Model With Dual Attention Mechanism

Jing Zhang, Tao Wen, Tao He, Xiangzhou Wang, Ruqian Hao, Juanxiu Liu, Xiaohui Du, Lin Liu; Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops, 2022, pp. 2096-2103

Abstract

The human stools are directly related to the health of human gastrointestinal function. Preliminary classification of the shape and colour of stools can diagnose the health status of peoples, therefore automatic recognition of stools is the current development direction of smart toilets. Due to the difficulty in identification with complex image content, this paper proposed a convolutional neural network called StoolNet to solve the current challenges. The architecture of StoolNet is based on ResNet and contains two output branches which perform colour and shape recognition, respectively. To improve the recognition performance, the dual attention mechanism was introduced into feature extraction stage. The accuracy value of our proposed model could achieve 99.7% and 94.4% for color and shape recognition on our test set, respectively. Experimental results show that, compared with other stool classification algorithms, our method possesses better capability of category discrimination on real dataset.

Related Material

[pdf]
[bibtex]
@InProceedings{Zhang_2022_CVPR, author = {Zhang, Jing and Wen, Tao and He, Tao and Wang, Xiangzhou and Hao, Ruqian and Liu, Juanxiu and Du, Xiaohui and Liu, Lin}, title = {Human Stools Classification for Gastrointestinal Health Based on an Improved ResNet18 Model With Dual Attention Mechanism}, booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) Workshops}, month = {June}, year = {2022}, pages = {2096-2103} }