From 5c88e483cf7289edc3b53631ca94bd7e090e4e92 Mon Sep 17 00:00:00 2001
From: Ubuntu <ubuntu@ip-172-16-40-167.us-gov-west-1.compute.internal>
Date: Mon, 14 Oct 2019 17:54:09 +0000
Subject: [PATCH] check length of sim_images

---
 cufacesearch/cufacesearch/searcher/output_mapping.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/cufacesearch/cufacesearch/searcher/output_mapping.py b/cufacesearch/cufacesearch/searcher/output_mapping.py
index 1caad17..691952a 100644
--- a/cufacesearch/cufacesearch/searcher/output_mapping.py
+++ b/cufacesearch/cufacesearch/searcher/output_mapping.py
@@ -187,9 +187,12 @@ def format_output(self, dets, sim_images, sim_dets, sim_score, options_dict=dict
 
     else:
       # no detections, dets contains list of query images sha1
+      if len(sim_images) != len(dets):
+        print("Did not get enough results! {} vs {}".format(len(sim_images), len(dets)))
 
       for i in range(len(dets)):
         sha1 = dets[i][0]
+        print("Preparing output for input {} with SHA1 {}".format(i, sha1))
 
         # Add one output for each image query
         output.append(dict())
@@ -199,7 +202,7 @@ def format_output(self, dets, sim_images, sim_dets, sim_score, options_dict=dict
           output[out_i][self.map['query_url']] = dets[i][1]
 
         nb_images = 0
-        if sim_images[i] and sim_images[i][0]:
+        if i<len(sim_images) and sim_images[i] and sim_images[i][0]:
           nb_images = len(sim_images[i][0])
 
         output[out_i][self.map['similar_images']] = OrderedDict([[self.map['number_images'], nb_images],