An Improved Detection Algorithm for Ischemic Stroke NCCT Based on YOLOv5

Abstract

Cerebral stroke (CS) is a heterogeneous syndrome caused by multiple disease mechanisms. Ischemic stroke (IS) is a subtype of CS that causes a disruption of cerebral blood flow with subsequent tissue damage. Noncontrast computer tomography (NCCT) is one of the most important IS detection methods. It is difficult to select the features of IS CT within computational image analysis. In this paper, we propose AC-YOLOv5, which is an improved detection algorithm for IS. The algorithm amplifies the features of IS via an NCCT image based on adaptive local region contrast enhancement, which then detects the region of interest via YOLOv5, which is one of the best detection algorithms at present. The proposed algorithm was tested on two datasets, and seven control group experiments were added, including popular detection algorithms at present and other detection algorithms based on image enhancement. The experimental results show that the proposed algorithm has a high accuracy (94.1% and 91.7%) and recall (85.3% and 88.6%) rate; the recall result is especially notable. This proves the excellent performance of the accuracy, robustness, and generalizability of the algorithm.