Here, I've implemented Full-Stack Multi-Class Image Classification in PyTorch. I've used the following Tech Stack:
- Frontend: NextJS + TailwindCSS,
- Backend: FastAPI,
- ML Algorithm: CNN,
Install all the dependencies using:
First go to backend > server:
pip install -r requirements.txt
uvicorn main:app --reload
Then go to client:
yarn
yarn dev
Some previews