Sign inSign up

cvachet/object-detection-yolos-api

By cvachet

•Updated over 1 year ago

Object detection API using YOLOS transformer based models and FastAPI

Image
API management
Machine learning & AI
0

1.1K

cvachet/object-detection-yolos-api repository overview

⁠Object detection with YOLOS model and FastAPI

Aim: AI-driven object detection via FastAPI using YOLOS transformer-based models (trained on COCO image dataset)

ML models: yolos-tiny, yolos-small

⁠Github project and documentation

URL: https://github.com/clementsan/object_detection_yolos_api⁠

⁠Execution via docker container

⁠1. Download docker image and run docker container

Command lines:

docker pull cvachet/object-detection-yolos-api

docker run --name object-detection-yolos-api-cont -p 8000:8000 object-detection-yolos-api

⁠2. Run query via API

Command lines:

  • Check Endpoint "/":

curl -X 'GET' 'http://0.0.0.0:8000/⁠' -H 'accept: application/json'

  • Check Endpoint "/api/v1/info":

curl -X 'GET' 'http://0.0.0.0:8000/info⁠' -H 'accept: application/json'

  • Call Endpoint "/api/v1/detect":

curl -X 'POST' -F "image=@./tests/data/savanna.jpg" http://0.0.0.0:8000/api/v1/detect⁠

Tag summary

Content type

Image

Digest

sha256:61da4a06b…

Size

338.9 MB

Last updated

over 1 year ago

docker pull cvachet/object-detection-yolos-api