基于elasticsearch的图片相似搜索

在git上发现有类似的项目
https://github.com/ascribe/image-match/blob/master/docs/source/start.rst
帮助文档地址是
https://image-match.readthedocs.io/en/latest/start.html。

需要先建立索引

import osimport hashlibfrom elasticsearch import Elasticsearch, ConnectionError, RequestError, NotFoundErrorfrom image_match.elasticsearch_driver import SignatureESDOC_TYPE = 'image'MAPPINGS = {  "mappings": {    DOC_TYPE: {      "dynamic": True,      "properties": {        "metadata": {            "type": "object",            "dynamic": True,            "properties": {                "tenant_id": { "type": "string", "index" : "not_analyzed"   }            }        }      }    }  }}def setup_index():    es = Elasticsearch()    try:        es.indices.create(index="images",  body=MAPPINGS)    except RequestError as e:        if e.error == u'index_already_exists_exception':            es.indices.delete(index_name)        else:            raisesetup_index()

然后dump数据

from elasticsearch import Elasticsearchfrom image_match.elasticsearch_driver import SignatureESes = Elasticsearch()ses = SignatureES(es)ses.add_image('https://upload.wikimedia.org/wikipedia/commons/thumb/e/ec/Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg/687px-Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg')ses.add_image('https://pixabay.com/static/uploads/photo/2012/11/28/08/56/mona-lisa-67506_960_720.jpg')ses.add_image('https://upload.wikimedia.org/wikipedia/commons/e/e0/Caravaggio_-_Cena_in_Emmaus.jpg')ses.add_image('https://c2.staticflickr.com/8/7158/6814444991_08d82de57e_z.jpg')

索引结果如下

然后进行检索

ses.search_image('https://pixabay.com/static/uploads/photo/2012/11/28/08/56/mona-lisa-67506_960_720.jpg')

返回结果如下， dist越小越好。

[ {'dist': 0.0,  'id': u'AVM37oZq0osmmAxpPvx7',  'metadata': None,  'path': u'https://pixabay.com/static/uploads/photo/2012/11/28/08/56/mona-lisa-67506_960_720.jpg',  'score': 7.937254}, {'dist': 0.22095170140933634,  'id': u'AVM37nMg0osmmAxpPvx6',  'metadata': None,  'path': u'https://upload.wikimedia.org/wikipedia/commons/thumb/e/ec/Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg/687px-Mona_Lisa,_by_Leonardo_da_Vinci,_from_C2RMF_retouched.jpg',  'score': 0.28797293}, {'dist': 0.42557196987336648,  'id': u'AVM37p530osmmAxpPvx9',  'metadata': None,  'path': u'https://c2.staticflickr.com/8/7158/6814444991_08d82de57e_z.jpg',  'score': 0.0499953}]