Skip to content

API for object recognition via POST or GET request of an image URL, can be set with different models.

Notifications You must be signed in to change notification settings

dvidd/api-photo-ai

Repository files navigation

api-photo-ai

Api for object recognition via POST or GET request of an image url, using the model MobileNetSSD

Example

input :

Output : ['bottle: 61.01%', 'cat: 69.77%', 'person: 92.28%']

How to used to api

curl  "https://localhost:5000/api?{[URL]}"

It would return a json array with the objects detected and the probability of them

If you don't specify the model or change the model it would use by default MobileNetSSD

Supported models

Model Description
MobileNetSSD The mobilenet-ssd model is a Single-Shot multibox Detection (SSD) network intended to perform object detection. This model is implemented using the Caffe* framework. For details about this model, check out the repository.

How make it work local :

To run the server first install the requierements.txt, then run

$ python3 server.py

In server.py there is a list of object that with using the MobileNetSSD can recognize, you can change the model and the list.

To check that it works you can run :

$ python3 tests/tests.py

TODO :

  • do support to others models
  • add to a server so the people can use it

About

API for object recognition via POST or GET request of an image URL, can be set with different models.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published