Ejemplo n.º 1
0
    # method = Edge()
    # samples = method.make_samples(db)
    # query = method.histogram(linkInput)
    # result = inferInput(query, samples=samples, depth=depth, d_type=d_type)
    # print(result)

    # # retrieve by gabor
    # method = Gabor()
    # samples = method.make_samples(db)
    # query = samples[query_idx]
    # _, result = infer(query, samples=samples, depth=depth, d_type=d_type)
    # print(result)

    # #retrieve by HOG
    method = HOG()
    samples = method.make_samples(db)
    query = method.get_featInput(img)
    start_time = time.time()

    result = inferInput(query, samples=samples, depth=depth, d_type=d_type)
    end_time = time.time()
    print('total run-time: %f ms' % ((end_time - start_time) * 1000))
    print(result)

    # retrieve by VGG
    method = VGGNetFeat()
    samples = method.make_samples(db)
    query = method.get_featInput(img)
    start_time = time.time()

    result = inferInput(query, samples=samples, depth=depth, d_type=d_type)