Deep Learning for Automated Wound Classification And Segmentation (2408.11064v1)

Abstract: Wounds, such as foot ulcers, pressure ulcers, leg ulcers, and infected wounds, come up with substantial problems for healthcare professionals. Prompt and accurate segmentation is crucial for effective treatment. However, contemporary methods need an exhaustive model that is qualified for both classification and segmentation, especially lightweight ones. In this work, we tackle this issue by presenting a new architecture that incorporates U-Net, which is optimized for both wound classification and effective segmentation. We curated four extensive and diverse collections of wound images, utilizing the publicly available Medetec Dataset, and supplemented with additional data sourced from the Internet. Our model performed exceptionally well, with an F1 score of 0.929, a Dice score of 0.931 in segmentation, and an accuracy of 0.915 in classification, proving its effectiveness in both classification and segmentation work. This accomplishment highlights the potential of our approach to automating wound care management.