A Multi-Task Attention-Driven SegNet for Lung Infection Segmentation and Classification From HRCT Images

Also Available Domains Machine Learning

Project Code :TCMAPY2504

Objective

This project builds a multi-task deep learning framework using MS-CSTA-Net and DualMask-SegNet v5.4 to jointly segment and classify lung infections from HRCT images via multi-scale attention and dual mask learning. A web interface allows image upload, segmentation map viewing, and classification results, with performance evaluated using accuracy, precision, recall, F1-score, and IoU. The scalable system adapts to diverse datasets, reduces manual effort, and offers interpretable infection visualizations to support medical imaging research.

Abstract

The proposed system presents a multi-task framework for automated lung infection analysis using high-resolution computed tomography (HRCT) images. It integrates two novel deep learning architectures: MS-CSTA-Net for multi-class classification of lung infections and DualMask-SegNet v5.4 for precise segmentation of affected regions. The classification module employs a multi-scale cross-scale token attention mechanism to capture both global and local features, enhancing recognition of subtle infection patterns. The segmentation module leverages dual mask learning to accurately delineate infection boundaries, improving interpretability and feature extraction for downstream classification tasks. The system is built on a Flask-based framework, offering a web interface with modules for Home, Register, Login, Prediction or Classification, and Logout. Users can input HRCT images, and the system provides segmentation maps alongside predicted infection classes. Experimental evaluation demonstrates improved performance in terms of accuracy, precision, recall, and F1-score, highlighting the model’s capability to handle complex pulmonary image patterns. The framework offers an efficient and scalable solution for lung infection analysis, combining robust feature extraction, attention mechanisms, and dual-task learning in a single unified system.

Keywords: HRCT images, lung infection, segmentation, classification, MS-CSTA-Net, DualMask-SegNet, attention mechanism, deep learning, feature extraction, multi-task learning.

NOTE: Without the concern of our team, please don't submit to the college. This Abstract varies based on student requirements.

Block Diagram

Specifications

5.2 HARDWARE REQUIREMENTS

β€’        Processor                                - I5/Intel Processor

β€’        RAM                                       - 8GB (min)

β€’        Hard Disk                                - 160 GB

β€’        Key Board                               - Standard Windows Keyboard

β€’        Mouse                                      - Two or Three Button Mouse

β€’        Monitor                                    - Any

5.3 SOFTWARE REQUIREMENS

β€’        Operating System                   :  Windows 7/8/10

β€’        Server side Script                   :  HTML, CSS, Bootstrap & JS

β€’        Programming Language         :  Python

β€’        Libraries                                 :  Flask, Pandas, Mysql. connector, Os, Numpy, Scikit- learn, sklearn, Preprocessor, tensor flow, keras, roboflow                                                    

β€’         IDE/Workbench                     :  VS-Code

β€’        Technology                             :  Python 3.10+

β€’        Server Deployment                 :  Xampp Server

β€’        Database                                 :  MySQL

Demo Video

mail-banner
call-banner
contact-banner
Request Video