Skip to content

PasPelle/ML_vs_Cancer

Repository files navigation

TEAM

Daniel Stark: Creative director, boasting more than eight years in the e-commerce industry, fervently interested in the incorporation of technologies into marketing strategies.
kristjanqarri kristjanqarri
Katažyna Balcevič: Experienced Android developer with a strong four-year background in the field, passionate about integrating expertise in ML Engineering with a dedication to transforming the healthcare industry.
kristjanqarri kristjanqarri
Kristjan Qarri: Medical Technology Engineer & passionate about the application of technology in healthcare.
kristjanqarri kristjanqarri
Pasquale Pellegrini (Team Leader): Data scientist with a background in cancer research and neurodegeneration.
kristjanqarri kristjanqarri

BACKGROUND

This project aims to leverage the pre-trained VGG16 Convolutional Neural Network on the ImageNet database. In addition dense and prediction layers are added to perform image classification in the healthy and metastasis classes.

WORKFLOW

  1. Selected a balanced subset of 2000 images from each class for training, with a 50:50 distribution.
  2. Pre-processed (normalized) images were then input into the VGG16 network.
  3. We developed an API using FastAPI and deployed the model as a web app using Streamlit.

APPLICATION

Our Streamlit app is a user-friendly platform designed for histopathologic cancer detection. We've tailored it to enable users to upload histopathologic images in specifically in 'tif' or 'tiff' file formats and receive accurate predictions regarding the presence of cancer. The app is enhanced with HTML styling for better visual appeal and offers an intuitive interface for ease of use. Key features of our app include:

  1. Image Analysis and Prediction: Users can request a prediction upon uploading an image. Our app uses a backend API to analyze the image and return a cancer prediction.
  2. Progress Feedback: We provide a progress bar and spinner, offering visual feedback during the image processing phase.
  3. Result Display with Accessibility in Mind: The app displays the prediction results in colors specifically chosen to be friendly to the visually impaired, distinguishing between malignant (displayed in a visually distinct red) and non-malignant results (shown in a clear green), accompanied by probability scores.
  4. Error Handling: We have integrated error messages for different scenarios, such as errors in prediction or API request failures.
  5. User-Friendly Design: Our use of HTML - styled text ensures clear and large font displays, enhancing the overall user experience.

LINKS

API: https://imageanalysis-ulbdrxxy6a-ew.a.run.app/docs
Web application: https://mlvscancer4.streamlit.app/
Image to test prediction: Image with Malignant tissue / Image with Non-Malignant tissue
note: Download the .TIFF image from the link above to test the prediction of the model

FUTURE PERSPECTIVES

  1. Extend the scope of our project to predict various types of primary cancers using a single urine test.
  2. Explore the potential of the existing models to detect different cancer types and implement necessary adaptations to accommodate diverse cancer profiles.
  3. Focus on creating a globally accessible primary cancer prediction application by incorporating language localization, adapting to different healthcare systems, and addressing specific regional healthcare needs and challenges.
  4. Develop educational resources and materials to raise awareness about the importance of early cancer detection. Conduct outreach programs to educate the public, healthcare professionals, and relevant stakeholders about the benefits and capabilities of the primary cancer prediction app.

CITATION

Will Cukierski. (2018). Histopathologic Cancer Detection. Kaggle. https://kaggle.com/competitions/histopathologic-cancer-detection

KAGGLE COMPETITION

https://www.kaggle.com/competitions/histopathologic-cancer-detection

About

Final project

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 4

  •  
  •  
  •  
  •