Object classification using PyTorch as a web service
See requirements.txt
To get a local copy up and running follow these simple steps.
- virtualenv
pip/pip3 install virtualenv
- Clone the repo
git clone https://github.com/tachillon/Object-Classification-PyTorch-Web-Service
- Create a virtualenv
python3 -m venv .venv
- Activate your virtualenv
source .venv/bin/activate
- Install requirements
pip3 install -r requirements.txt
- Build the Docker container
docker build -t <container_name>:<tag> .
- Regular usage
python3 classificator.py
- Using Docker
docker run --rm -p 8080:8080 <container_name>:<tag>
- Do the POST request
POST request http://localhost:8080/classify*
*with the image encoded as base64
Contributions are what make the open source community such an amazing place to be learn, inspire, and create. Any contributions you make are greatly appreciated.
- Fork the Project
- Create your Feature Branch (
git checkout -b feature/AmazingFeature
) - Commit your Changes (
git commit -m 'Add some AmazingFeature'
) - Push to the Branch (
git push origin feature/AmazingFeature
) - Open a Pull Request
Distributed under the MIT License. See LICENSE
for more information.
Achille-Tâm GUILCHARD - achilletamguilchard@gmail.com
Project Link: https://github.com/tachillon/Object_Classification_PyTorch_Web_Service