A Method for Expanding the Training Set of White Blood Cell Images.

Abstract

In medicine, the count of different types of white blood cells can be used as the basis for diagnosing certain diseases or evaluating the treatment effects of diseases. The recognition and counting of white blood cells have important clinical significance. But the effect of recognition based on machine learning is affected by the size of the training set. At present, researchers mainly rely on image rotation and cropping to expand the dataset. These methods either add features to the white blood cell image or require manual intervention and are inefficient. In this paper, a method for expanding the training set of white blood cell images is proposed. After rotating the image at any angle, Canny is used to extract the edge of the black area caused by the rotation and then fill the black area to achieve the purpose of expanding the training set. The experimental results show that after using the method proposed in this paper to expand the training set to train the three models of ResNet, MobileNet, and ShuffleNet, and comparing the original dataset and the method trained by the simple rotated image expanded dataset, the recognition accuracy of the three models is obviously improved without manual intervention.