File size: 734 Bytes
708f5d3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import configparser
import search_engine_model as sem
def main():
config_manager_obj = configparser.ConfigParser()
config_manager_obj.read('./config.cfg')
image_root_dir = config_manager_obj['SEARCH_ENGINE']['image_root_dir']
csv_file_path = config_manager_obj['SEARCH_ENGINE']['csv_file_path']
model_manager_obj = sem.SearchEngineModel(image_root_dir, csv_file_path)
search_results = model_manager_obj.search_image_by_text_prompt('Cat', 10)
print('Search results found by text prompt -> ', search_results)
search_results = model_manager_obj.search_image_by_image_prompt('./dataset/000000001675.jpg', 10)
print('Search results found by image prompt -> ', search_results)
main() |