| 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() |