def create_and_save(name, images, file_path):
index = imsearch.init(name)
index.cleanIndex()
index.addImageBatch(images)
index.createIndex()
index.saveIndex(file_path)
index.cleanIndex()
def test_init(self, mock_index, config_init_mock):
instance = mock_index()
instance.name = 'test'
index = imsearch.init('test', MONGO_URI='mongodb://localhost:27017/')
self.assertEqual(index, instance)
self.assertEqual(index.name, 'test')
config_init_mock.assert_called_with(
{'MONGO_URI': 'mongodb://localhost:27017/'})
def create_index_with_config(name):
'''
parameters:
name: name of the index (unique identifier name)
MONGO_URI
REDIS_URI
DETECTOR_MODE: select 'local' or 'remote'.
local: detector backend should be running on the same machine
remote: Process should not start detector backend
pass any configuration you want to expose as environment variable.
'''
index = imsearch.init(name=name,
MONGO_URI='mongodb://*****:*****@123.456.78.111:6379/0',
DETECTOR_MODE='local')
def create_index(name, images):
# Initialize the index
index = imsearch.init(name)
# Clear the index and data if any exists
index.cleanIndex()
# Add single image to index (image path locally stored)
index.addImage(images[0])
# Add image using URL with same interface
index.addImage(
"https://www.wallpaperup.com/uploads/wallpapers/2014/04/14/332423/d5c09641cb3af3a18087937d55125ae3-700.jpg")
# Add images in batch (List of image paths locally stored)
index.addImageBatch(images[1:])
# Build the index
index.createIndex()
return index
import cv2
import glob
import os, sys
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
import imsearch
all_images = glob.glob(
os.path.join(os.path.dirname(__file__), '..', 'images/*.jpg'))
index = imsearch.init('test')
# index.cleanIndex()
# index.addImageBatch(all_images[0:25])
# index.addImage(all_images[25])
index.createIndex()
similar = index.knnQuery(all_images[25], policy='object')
qImage = imsearch.utils.load_image(all_images[25])
# similar = index.knnQuery('https://www.wallpaperup.com/uploads/wallpapers/2014/04/14/332423/d5c09641cb3af3a18087937d55125ae3-700.jpg', policy='object')
# qImage = imsearch.utils.load_image('https://www.wallpaperup.com/uploads/wallpapers/2014/04/14/332423/d5c09641cb3af3a18087937d55125ae3-700.jpg')
cv2.imshow('qImage', qImage)
for _i, _s in similar:
rImage = cv2.imread(_i['image'])
print([x['name'] for x in _i['primary']])
print(_s)
cv2.imshow('rImage', rImage)
cv2.waitKey(0)
'BUCKET_NAME': aws_config_file['BUCKET_NAME'],
'STORAGE_MODE': 's3'
}
def show_results(similar, qImage):
qImage = imsearch.utils.check_load_image(qImage)
qImage = cv2.cvtColor(qImage, cv2.COLOR_RGB2BGR)
cv2.imshow('qImage', qImage)
for _i, _s in similar:
rImage = cv2.cvtColor(imsearch.utils.check_load_image(_i['image']),
cv2.COLOR_RGB2BGR)
print([x['name'] for x in _i['primary']])
print(_s)
cv2.imshow('rImage', rImage)
cv2.waitKey(0)
if __name__ == "__main__":
all_images = glob.glob(
os.path.join(os.path.dirname(__file__), '..', 'images/*.jpg'))
index = imsearch.init(name='test', **aws_config)
index.cleanIndex()
index.addImageBatch(all_images)
index.createIndex()
# query with image URL
img_url = 'https://www.wallpaperup.com/uploads/wallpapers/2014/04/14/332423/d5c09641cb3af3a18087937d55125ae3-700.jpg'
similar = index.knnQuery(image_path=img_url, k=10, policy='global')
show_results(similar, img_url)
import cv2
import glob
import os, sys
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))
import imsearch
all_images = glob.glob(
os.path.join(os.path.dirname(__file__), '..', 'images/*.jpg'))
index = imsearch.init(
'test',
MONGO_URI='mongodb://*****:*****@216.165.71.213:6379/0')
# index.cleanIndex()
# index.addImageBatch(all_images[0:25])
# index.addImage(all_images[25])
index.createIndex()
# similar = index.knnQuery(all_images[25], policy='object')
# qImage = imsearch.utils.load_image(all_images[25])
similar = index.knnQuery(
'https://www.wallpaperup.com/uploads/wallpapers/2014/04/14/332423/d5c09641cb3af3a18087937d55125ae3-700.jpg',
policy='object')
qImage = imsearch.utils.load_image(
'https://www.wallpaperup.com/uploads/wallpapers/2014/04/14/332423/d5c09641cb3af3a18087937d55125ae3-700.jpg'
)
cv2.imshow('qImage', qImage)
