def main(task, num_docs): config() data_path = './MNIST_Data/test/t10k-images-idx3-ubyte' if task == 'index': workspace = os.environ['WORKDIR'] if os.path.exists(workspace): print( f'\n +---------------------------------------------------------------------------------+ \ \n | ������ | \ \n | The directory {workspace} already exists. Please remove it before indexing again. | \ \n | ������ | \ \n +---------------------------------------------------------------------------------+' ) f = Flow().load_config('mnist-index.yml') with f: f.index_ndarray(load_mnist(data_path), size=num_docs, batch_size=2) elif task == 'search': result_html = [] f = Flow().load_config('mnist-query.yml') with f: f.search_ndarray(load_mnist(data_path), size=num_docs, batch_size=1, output_fn=print_result, top_k=10) write_html('result.html') else: raise NotImplementedError( f'unknown task: {task}. A valid task is either `index` or `search`.' )
def main(task, num_docs): config() data_path = './MNIST_Data/test/t10k-images-idx3-ubyte' if task == 'index': f = Flow().load_config('mnist-index.yml') with f: f.index_ndarray(load_mnist(data_path), size=num_docs, batch_size=2) elif task == 'search': result_html = [] f = Flow().load_config('mnist-query.yml') with f: f.search_ndarray(load_mnist(data_path), size=num_docs, batch_size=1, output_fn=print_result, top_k=10) write_html('result.html') else: raise NotImplementedError( f'unknown task: {task}. A valid task is either `index` or `search`.' )