Build a multi class image classification model using any framework(PyTorch, Keras, etc..) and deploy via a REST API
- Fork or clone this repo
- Maintain the directory structure given for the project
- Use Python 3.7+
- If you need additional imports specify them in
requirements.txt
- Model should be trained on the given dataset.
- Create a model artifact and save it under
/models. - Report accuracy on
test.
- Serve the model as a REST API using FastAPI
- Be able to use CURL to send in text input and return the prediction.
The dataset you will be using contains 10 classes of apparels. You need to train basic image classification model which will classify given text into these 10 classes categories. You can find the dataset and details data format and labels can be found here
Dataset contains two splits train and test.
Timebox this challenge to 8-10 hours. After completing the assignment, please compress whole repo and send it.