Automatic body region localization in 3D-CT images based on the improved YOLO model

Abstract

Automatic body region localization in medical three-dimensional (3D)-CT images is a critical step of computerized body-wide Automatic Anatomy Recognition (AAR) system, which can be applied for radiotherapy planning and interest slices retrieving. Currently, the complex internal structure of human body and time consuming computation are the main challenges for the localization. Therefore, this paper introduces and improves the YOLO-v3 model into the body region localization for these problems. First, seven categories of body regions in a CT volume image I are defined based on the modification version of our previous work. Second, an improved YOLO-v3 model is trained to classify each axial slice into one of the seven categories. Then, the effectiveness of the proposed method is evaluated on 3D-CT images that collected from 220 subjects. The experimental results demonstrate that the slice localizing error is less than 3 NoS (Number of slices), which is competitive to the state-of-the-art methods. Beyond this, our method is simple and computationally efficient owing to its less training time, and the average computational time for localizing a volume CT images is about 3 second, which shows potential for a further application.