summaryrefslogtreecommitdiff
path: root/megapixels/app/server
diff options
context:
space:
mode:
Diffstat (limited to 'megapixels/app/server')
-rw-r--r--megapixels/app/server/api.py14
1 files changed, 10 insertions, 4 deletions
diff --git a/megapixels/app/server/api.py b/megapixels/app/server/api.py
index 2f78ecd3..bc60118c 100644
--- a/megapixels/app/server/api.py
+++ b/megapixels/app/server/api.py
@@ -58,6 +58,10 @@ def upload(name):
# get detection as BBox object
bboxes = detector.detect(im_np, largest=True)
+ if not len(bboxes):
+ return jsonify({
+ 'error': 'bbox'
+ })
bbox = bboxes[0]
dim = im_np.shape[:2][::-1]
bbox = bbox.to_dim(dim) # convert back to real dimensions
@@ -73,16 +77,18 @@ def upload(name):
distances, indexes = faiss_dataset.search(query, 10)
if len(indexes) == 0:
- print("weird, no results!")
- return []
+ return jsonify({
+ 'error': 'nomatch'
+ })
# get the results for this single query...
distances = distances[0]
indexes = indexes[0]
if len(indexes) == 0:
- print("no results!")
- return []
+ return jsonify({
+ 'error': 'nomatch'
+ })
lookup = {}
ids = [i+1 for i in indexes]