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`.'
)
