mirror of
https://github.com/wassname/IndicoIo-python.git
synced 2026-08-09 11:40:38 +08:00
Merge pull request #107 from IndicoDataSolutions/Chris/fer-detect-no-resize
FIX: FER No resize for detect
This commit is contained in:
@@ -27,6 +27,10 @@ def fer(image, cloud=None, batch=False, api_key=None, **kwargs):
|
||||
:type image: list of lists
|
||||
:rtype: Dictionary containing emotion probability pairs
|
||||
"""
|
||||
image = image_preprocess(image, batch=batch)
|
||||
|
||||
image = image_preprocess(image, batch=batch,
|
||||
size=None if kwargs.get("detect") else (48, 48)
|
||||
)
|
||||
|
||||
url_params = {"batch": batch, "api_key": api_key}
|
||||
return api_handler(image, cloud=cloud, api="fer", url_params=url_params, **kwargs)
|
||||
|
||||
@@ -41,7 +41,8 @@ def image_preprocess(image, size=(48,48), min_axis=None, batch=False):
|
||||
else:
|
||||
raise IndicoError("Image must be a filepath, base64 encoded string, or a numpy array")
|
||||
|
||||
out_image = resize_image(out_image, size, min_axis)
|
||||
if size or min_axis:
|
||||
out_image = resize_image(out_image, size, min_axis)
|
||||
|
||||
# convert to base64
|
||||
temp_output = StringIO.StringIO()
|
||||
@@ -60,7 +61,7 @@ def resize_image(image, size, min_axis):
|
||||
warnings.warn(
|
||||
"An aspect ratio greater than 10:1 is not recommended",
|
||||
Warning
|
||||
)
|
||||
)
|
||||
size_arr = [0,0]
|
||||
size_arr[min_idx] = min_axis
|
||||
size_arr[other_idx] = int(min_axis * aspect)
|
||||
@@ -89,4 +90,3 @@ def get_element_type(_list, dimens):
|
||||
for _ in xrange(len(dimens)):
|
||||
elem = elem[0]
|
||||
return type(elem)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user