Link: https://www.youtube.com/watch?v=QUOWaZT4XEk&feature=youtu.be
This is a simple image semantic segmentation problem streamlit app trained on the top of Tensorflow API. The trained models i.e binary model,multilabel model, 4 image segmentation models combinally takes an image (Defected Steel Image) as an input and predict the class of defect from defect1, defect2, defect3, defect4 with defected area pixels highlighted.
While working for one IT company for a manufacturing client, there they were using compuer vision cameras to catch the steel images and they have their algorithm installed to find the defected images but that was not based on machine learning or deep learning, so i thought why can't i create one solution to keep in my mind about the industry business problem.
This project is divided into two part:
- Training a deep learning model using Tensorflow2.0 using gradient tape.
- Building and hosting a Streamlit web app on Heroku.
- A user can choose image from a device or capture it using a pre-built camera.
- Used Amazon S3 Bucket to store the uploaded image and predictions.
- After uploading the image, the predictions are displayed on a Bar Chart.
Lucky Chauhan |
- Google Images Download - This project wouldn't have been possible without this tool. It saved my enormous amount of time while collecting the data. A huge shout-out to its creator Hardik Vasa.