如何解决 KeyError：'img'？

Question

I'm trying the content based image retrieval from https://github.com/pochih/CBIR . I used the short code below to try retrieve images based on color histogram.

from evaluate import infer
from six.moves import cPickle
import os

CACHE_DIR = 'cache'
SAMPLE_CACHE = 'histogram_cache-region-n_bin12-n_slice3' # Change this to your cache
NUM_IMAGES = 10

samples = cPickle.load(open(os.path.join(CACHE_DIR, SAMPLE_CACHE), "rb"))

query = samples[30]

ap, res = infer(query, samples, db=None, sample_db_fn=None, depth=NUM_IMAGES, d_type='d1')

print('Query: {}'.format(query['img']))

print ('Top {} similar images: '.format(NUM_IMAGES))

for result in res:
  print(result['img'])

However, I got the error below:

Query: database\bawang\bawang_126.jpg
Top 10 similar images: 
Traceback (most recent call last):
  File "C:/Users/HP/PycharmProjects/USM/CBIRR/src/retrieve.py", line 20, in <module>
    print(result['img'])
KeyError: 'img'

Can someone tell me how to fix this error?

Answer 1

You will never "solve" this because your "result" doesn't contain img attribute, only the following:

• cls
• dis

You must use data either from the cls attribute, that is result.cls (try to print vars(result.cls) ) or ap object (which I don't know anything about so take it with grain of salt).

Also, you can't count on the fact that results will be ordered like samples because results are sorted by the distance dis .