TE-YOLOF: Tiny and efficient YOLOF for blood cell detection
Fanxin Xu,Xiangkui Li,Hang Yang,Yali Wang,Wei Xiang
DOI: https://doi.org/10.1016/j.bspc.2021.103416
IF: 5.1
2022-03-01
Biomedical Signal Processing and Control
Abstract:Blood cell detection in microscopic images is an essential branch of medical image processing research. The research and application of computer vision algorithms in this field are more concerned about the trade-off between accuracy and model complexity. The FED detector modified by YOLOv3 is a representative light-weight model to detect blood cell objects such as red blood cells, white blood cells and platelets. But the detection precision of red blood cells in the FED model is relatively low compared with platelets and white blood cells due to the imbalance distribution of different types of cells. In this research, we propose a light-weight model based on YOLOF to address the relatively low precision of red blood cell detection problem, in order to achieve the overall improvement of detection precision. This object detector is called TE-YOLOF, Tiny and Efficient YOLOF. Model light-weighting is accomplished with the excellent feature extraction capabilities of EfficientNet as backbone and the ability of the Depthwise Separable Convolution to reduce the number of parameters while maintaining precision. Furthermore, the Mish activation function is employed to increase the precision. Extensive experiments on the BCCD dataset prove the effectiveness of the proposed model, which can achieve higher precision with less parameters than FED. TE-YOLOF is also effective on other cross-domain blood cell detection experiments.
engineering, biomedical
What problem does this paper attempt to address?