一种基于YOLO-V5和U-Net的全自动心外膜脂肪组织提取系统

摘要

本发明涉及一种基于YOLO‑V5和U‑Net的全自动心外膜脂肪组织提取系统，所述系统包括用于检测心脏空间位置的YOLO‑V5模型和用于分割心脏和提取心外膜脂肪组织的U‑Net模型。本发明还涉及一种基于YOLO‑V5和U‑Net的全自动心外膜脂肪组织提取系统的搭建及训练流程方法。本系统解决了U‑Net应U‑Net负样本较多情况下分割能力下降的问题及应用于大图片的小物体分割中的低效问题用于大图片的小物体分割中的低效问题，通过YOLO‑V5模型检测心脏位置和U‑Net模型分割心脏轮廓，使心外膜脂肪组织只需在心脏内通过脂肪阈值提取即可获取，提高了心外膜脂肪组织的提取精度，简化了U‑Net完成心外膜脂肪组织分割的计算量。