fix display bug for bit=8,16

Author: jiuntian

inference/indexer.py

@@ -45,6 +45,7 @@ def read_log_path(self):
         self.config = json.load(open(self.log_path + '/config.json'))
         self.nbit = self.config['arch_kwargs']['nbit']
         self.nclass = self.config['arch_kwargs']['nclass']
+        assert self.nbit % 8 == 0, 'Number of bit must be multiple of 8'
 
     def load_model(self):
         model_path = os.path.join(self.log_path, 'models', 'best.pth')

inference/web.py

@@ -37,7 +37,7 @@ def process_query(f, indexer: Indexer):
 
     time_string = f'Time taken: {(end_time - start_time):.3f}s\n'
     code_string = "".join(str(np.unpackbits(query_code)).split())[1:-2]
-    code_string = '\n'.join(code_string[i:i + 32] for i in range(0, len(code_string), 32))
+    code_string = '\n'.join(code_string[i:i + 8] for i in range(0, len(code_string), 8))
     return dist, img_paths, code_string, encoded_img_data, img_, time_string