Detection of Different Stages of Lungs Cancer in CT-Scan Images Using CNN on Embedded Systems
After my BSc degree was finished, I still had unquenched thirst for more knowledge within the AI and ML field. I enrolled in an AI course at the University of Essex. I worked on a some interesting projects there, however very little involved imagery, and so when it came to picking a thesis, I opted for a project that concerned imagery. The title I landed on was “Detection of Different Stages of Lungs Cancer in CT-Scan Images Using CNN on Embedded Systems”.
During this project I developed a model architecture for lightweight predictions and training, as the aim was to have the model run on a Odroid XU4, which is an embedded device development board for lightweight applications. This model was trained on the LIDC-IDRI lung cancer imaging dataset which was used for the entirety of the project. The result of the training was compared with the ResNet, MobileNet and VGG16 models as they are comparable lightweight models. Due to the time limitations of the project, the image sizes used, and training time was reduced. This is likely to have had a negative effect on the results, however, training all the models for one epoch with the current training data takes approximately 11 hours, and thus increasing the number of epochs to train will cost a lot of time. Despite this, the results of the project was very acceptable, netting me a top grade on the dissertation.