Summary
An Image classifier that classify whether an image is of Mamooty or lalettan
You can get the dataset by scraping google. You can use selenium automation as mentioned in this article or any other methods (like UrlLib2 + BeautifulSoup)
Elements
- We need dataset containing images of Mammooty and Mohanlal collected from google or some other source.
- We need Machine learning model (Classic ML algorithms, CNNs, RNNs, whatever...)
- We need Flask web app to make the project as an API (You can use fast api or the micro framework you prefer )
- Collect image dataset using selenium automation or some other method
- Build the machine learning/ deep learning model in your favourite framework
- Train the model
- Write a small flask web app to host
- Test the AI model and deploy
Possible users journey
- A user should be able to pass an image in a web interface or as GET/POST request with url as parameter.
- A use gets the human-readable classification as a response - like "Lalettan" or "Mammukka" or "I'm confused. Who is this guy?"
Acceptance criteria
- Its fine if the model is not performing that great!
- Should be able to send request to the API with image or image url as parameter and get classification result.
- Github repo with train, test and dataset preprocessing codes.