Automatic Detection and Segmentation of Ovarian Cancer Using a Multitask Model in Pelvic CT Images.

Journal: Oxidative medicine and cellular longevity
Published Date:

Abstract

Ovarian cancer is one of the most common malignant tumours of female reproductive organs in the world. The pelvic CT scan is a common examination method used for the screening of ovarian cancer, which shows the advantages in safety, efficiency, and providing high-resolution images. Recently, deep learning applications in medical imaging attract more and more attention in the research field of tumour diagnostics. However, due to the limited number of relevant datasets and reliable deep learning models, it remains a challenging problem to detect ovarian tumours on CT images. In this work, we first collected CT images of 223 ovarian cancer patients in the Affiliated Hospital of Qingdao University. A new end-to-end network based on YOLOv5 is proposed, namely, YOLO-OCv2 (ovarian cancer). We improved the previous work YOLO-OC firstly, including balanced mosaic data enhancement and decoupled detection head. Then, based on the detection model, a multitask model is proposed, which can simultaneously complete the detection and segmentation tasks.

Authors

  • Xun Wang
    College of Computer Science and Technology, China University of Petroleum, Dongying, China.
  • Hanlin Li
    Tianjin Institute of Urology, the Second Hospital of Tianjin Medical University, Tianjin 300211, China.
  • Pan Zheng
    Information Systems, University of Canterbury, Christchurch, New Zealand.