blob: 0cf56a318b62e8370ffe4dddef1af8218099fd65 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
|
import click
from app.search.params import Params
from app.search.search_dense import find_dense_embedding_for_images
@click.command('')
@click.option('-i', '--input', 'opt_fp_in', required=True,
help='Path to input image')
@click.pass_context
def cli(ctx, opt_fp_in):
"""
Search for an image (class vector) in BigGAN using gradient descent
"""
params = Params(opt_fp_in)
find_dense_embedding_for_images(params)
|