Improved YOLOv8-C2fCA for embryonic cell detection and counting
摘要
Selecting the right embryos for transfer is crucial to improving the success rate of In Vitro Fertilization and Embryo Transfer (IVF-ET). The precise identification and quantification of embryonic cells in the image is the key to the selection of transplanted embryos. However, the variability of the cells’ appearance and their overlap in the images pose a challenge to embryologists. To improve the accuracy and consistency of embryo selection, this paper proposes an improved YOLOv8, YOLOv8-C2fCA, which strategically embeds a Coordinate Attention (CA) into the Cross Stage Partial with Two Fusion (C2f) module of its backbone and neck networks. The resulting C2fCA module significantly improves the model’s ability to extract detailed features from overlapping cell structures. Compared with baseline and attention-based YOLOv8 models, YOLOv8-C2fCA achieved superior results on both private and public datasets, demonstrating higher accuracy, lower computational complexity, and faster inference speed, with only a minimal increase of 0.096 M in the number of parameters over the original YOLOv8. This model can be an effective tool for automated detection and counting of embryonic cells, significantly reducing clinicians’ workload.