mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-06-27 16:10:34 +08:00
Chris Lee
29 lines
1.2 KiB
Python
29 lines
1.2 KiB
Python
import requests
|
|
|
|
from indicoio.utils.image import image_preprocess
|
|
from indicoio.utils.api import api_handler
|
|
|
|
def image_recognition(image, cloud=None, batch=False, api_key=None, version=None, **kwargs):
|
|
"""
|
|
Given an input image, returns a dictionary of image classifications with associated scores
|
|
|
|
* Input can be either grayscale or rgb color and should either be a numpy array or nested list format.
|
|
* Input data should be either uint8 0-255 range values or floating point between 0 and 1.
|
|
* Large images (i.e. 1024x768+) are much bigger than needed, minaxis resizing will be done internally to 144 if needed.
|
|
* For ideal performance, images should be square aspect ratio but non-square aspect ratios are supported as well.
|
|
|
|
Example usage:
|
|
|
|
.. code-block:: python
|
|
|
|
>>> from indicoio import image_recognition
|
|
>>> features = image_recognition(<filename>)
|
|
|
|
:param image: The image to be analyzed.
|
|
:type image: str
|
|
:rtype: dict containing classifications
|
|
"""
|
|
image = image_preprocess(image, size=144, min_axis=True, batch=batch)
|
|
url_params = {"batch": batch, "api_key": api_key, "version": version}
|
|
return api_handler(image, cloud=cloud, api="imagerecognition", url_params=url_params, **kwargs)
|