University of California - Santa Cruz
Student: Alex Salman
Last modified: June 3, 2022
This project proposes an architecture for classifying wound healing stages of a series of wound images. We generate a series of consecutive wound image frames and feed them to a 2D convolutional neural network combined with long short-term memory unit and a 3D convolutional neural network to learn spatio-temporal features associated with the healing trajectory. We also visualize the saliency maps to identify features the model is extracting. Both models can extract visual features related to wound healing and have relatively high classification accuracy.
- Generate videos from images using openCV
- Pad different length generated videos to have the same length (16 frames)
- Name vidoe after the name of the last frame appended
- Normalize array of frames to values between 0-1
- Split data to 75% training, 12.5% validation, and 12.5% test
- Learn spatio-temporal features using 2D-CNN+LSTM and 3D-CNN
- Has two versions for each model for high and low resolution images
- Visualize features extracted using saliency maps
Hemostasis, Inflammatory, Proliferative, and Maturation
See the following .ipynb files in this reposetory:
Installing required packages
pip3 install -r requirements.txt