Skip to main content Skip to main navigation menu Skip to site footer
Published: 2023-11-27

An automated approach for the kidney segmentation and detection of kidney stones on computed tomography using YOLO algorithms

Department of Electrical and Electronic Engineering, Sylhet Engineering College, Sylhet, 3100, Bangladesh
Department of Health Management Information System, Mymensingh Medical College Hospital, Mymensingh-2200, Bangladesh
Department of Electrical and Electronic Engineering, Sylhet Engineering College, Sylhet, 3100, Bangladesh
Department of Electrical and Electronic Engineering, Sylhet Engineering College, Sylhet, 3100, Bangladesh
Department of Electrical and Electronic Engineering, Sylhet Engineering College, Sylhet, 3100, Bangladesh
Department of Far, Power and Machinery, Sylhet Agricultural University, Sylhet-3100M, Bangladish
Department of Food Engineering and Technology, Sylhet Agricultural University, Sylhet-3100 , Bangladesh


Background: For effective diagnosis and treatment planning, accurate segmentation of the kidneys and detection of kidney stones are crucial. Traditional procedures are time-consuming and subject to observer variation. This study proposes an automated method employing YOLO algorithms for renal segmentation and kidney stone detection on CT scans to address these issues.

Methods: The dataset used in this study was sourced from the GitHub. The dataset contains a total of 1799 images, with 790 images labeled as 'containing kidney stones' and 1009 images labeled as 'not containing kidney stones'. U-Net architecture was utilized to precisely identify the region of interest, while YOLOv5 and YOLOv7 architecture was utilized to detect the stones. In addition, a performance comparison between the two YOLO models and other contemporary relevant models has been conducted.

Results: We obtained a kidney segmentation IOU of 91.4% and kidney stone detection accuracies of 99.5% for YOLOv7 and 98.7% for YOLOv5. YOLOv5 and YOLOv7 outperform the best existing models, including CNN, KNN, SVM, Kronecker CNN, Xresnet50, VGG16, etc. YOLOv7 possesses superior accuracy than YOLOv5. The only issue we encountered with the YOLOv7 model was that it demanded more training time than the YOLOv5 model.

Conclusion: The results demonstrate that the proposed AI-based method has the potential to improve clinical procedures, allowing radiologists and urologists to make well-informed decisions for patients with renal pathologies. As medical imaging technology progresses, the incorporation of deep learning techniques such as YOLO holds promise for additional advances in automated diagnosis and treatment planning.


Download data is not yet available.


  1. Power SP, Moloney F, Twomey M, James K, O’Connor OJ, Maher MM. Computed tomography and patient risk: Facts, perceptions and uncertainties. World Journal of Radiology. 2016;8(12):902-15.
  2. Caglayan A, Horsanali MO, Kocadurdu K, Ismailoglu E, Guneyli S. Deep learning model-assisted detection of kidney stones on computed tomography. International Braz J Urol. 2022; 48:830-9.
  3. Fang W, Wang L, Ren P. Tinier-YOLO: A real-time object detection method for constrained environments. IEEE Access. 2019; 8:1935-44.
  4. Pacal I, Karaman A, Karaboga D, Akay B, Basturk A, Nalbantoglu U, Coskun S. An efficient real-time colonic polyp detection with YOLO algorithms trained by using negative samples and large datasets. Computers in Biology and Medicine. 2022; 141:105031.
  5. da Cruz LB, Araújo JD, Ferreira JL, Diniz JO, Silva AC, de Almeida JD, de Paiva AC, Gattass M. Kidney segmentation from computed tomography images using deep neural network. Computers in Biology and Medicine. 2020; 123:103906.
  6. Daudon M, Williams Jr JC. Characteristics of human kidney stones. Kidney Stones: Medical and Surgical Management. 2019; 30:77.
  7. Singh P, Granberg CF, Harris PC, Lieske JC, Licht JH, Weiss A, Milliner DS. Primary hyperoxaluria type 3 can also result in kidney failure: a case report. American Journal of Kidney Diseases. 2022;79(1):125-8.
  8. Nestler T, Haneder S, Hokamp NG. Modern imaging techniques in urinary stone disease. Current Opinion in Urology. 2019;29(2):81-8.
  9. Isensee F, Jaeger PF, Kohl SA, Petersen J, Maier-Hein KH. nnU-Net: a self-configuring method for deep learning-based biomedical image segmentation. Nature Methods. 2021;18(2):203-11.
  10. Wu X, Sahoo D, Hoi SC. Recent advances in deep learning for object detection. Neurocomputing. 2020;396:39-64.
  11. Zhang J, Xie Y, Wu Q, Xia Y. Medical image classification using synergic deep learning. Medical Image Analysis. 2019;54:10-9.
  12. Yildirim K, Bozdag PG, Talo M, Yildirim O, Karabatak M, Acharya UR. Deep learning model for automated kidney stone detection using coronal CT images. Computers in Biology and Medicine. 2021;135:104569.
  13. Almansour NA, Syed HF, Khayat NR, Altheeb RK, Juri RE, Alhiyafi J, Alrashed S, Olatunji SO. Neural network and support vector machine for the prediction of chronic kidney disease: A comparative study. Computers in Biology and Medicine. 2019; 109:101-11.
  14. Baygin M, Yaman O, Barua PD, Dogan S, Tuncer T, Acharya UR. Exemplar Darknet19 feature generation technique for automated kidney stone detection with coronal CT images. Artificial Intelligence in Medicine. 2022; 127:102274.
  15. Causey J, Stubblefield J, Qualls J, Fowler J, Cai L, Walker K, Guan Y, Huang X. An ensemble of u-net models for kidney tumor segmentation with CT images. IEEE/ACM Transactions on Computational Biology and Bioinformatics. 2021;19(3):1387-92.
  16. Jubayer F, Soeb JA, Mojumder AN, Paul MK, Barua P, Kayshar S, Akter SS, Rahman M, Islam A. Detection of mold on the food surface using YOLOv5. Current Research in Food Science. 2021; 4:724-8.
  17. Soeb MJ, Jubayer MF, Tarin TA, Al Mamun MR, Ruhad FM, Parven A, Mubarak NM, Karri SL, Meftaul IM. Tea leaf disease detection and identification based on YOLOv7 (YOLO-T). Scientific Reports. 2023;13(1):6078.
  18. Serrell EC, Best SL. Imaging in stone diagnosis and surgical planning. Current Opinion in Urology. 2022;32(4):397-404.
  19. Apena WO, Joseph OL. Convolutional neural network model layers improvement for segmentation and classification on kidney stone images using keras and tensorflow. Journal of Multidisciplinary Engineering Science and Technology. 2021;8(6):14151-56
  20. Aksakalli I, Kaçdioğlu S, Hanay Ys. Kidney x-ray images classification using machine learning and deep learning methods. Balkan Journal of Electrical and Computer Engineering. 2021;9(2):144-51.
  21. Verma J, Nath M, Tripathi P, Saini KK. Analysis and identification of kidney stone using K th nearest neighbour (KNN) and support vector machine (SVM) classification techniques. Pattern Recognition and Image Analysis. 2017;27:574-80.
  22. Patro KK, Allam JP, Neelapu BC, Tadeusiewicz R, Acharya UR, Hammad M, Yildirim O, Pławiak P. Application of Kronecker convolutions in deep learning technique for automated detection of kidney stones with coronal CT images. Information Sciences. 2023;640:119005.
  23. Akshaya M, Nithushaa R, Raja NS, Padmapriya S. Kidney stone detection using neural networks. In 2020 International Conference on System, Computation, Automation and Networking (ICSCAN) 2020 Jul 3 (pp. 1-4). IEEE.
  24. Vishmitha D, Yoshika K, Sivalakshmi P, Chowdary V, Shanthi KG, Yamini M. Kidney Stone Detection Using Deep Learning and Transfer Learning. In 2022 4th International Conference on Inventive Research in Computing Applications (ICIRCA) 2022 (pp. 987-992). IEEE.

How to Cite

Rabby S, Hossain F, Das S, Rahman I, Das S, Soeb J, Jubaye MF. An automated approach for the kidney segmentation and detection of kidney stones on computed tomography using YOLO algorithms. jidhealth [Internet]. 2023 Nov. 27 [cited 2024 Feb. 23];6(4):963-70. Available from: