Ejemplo n.º 1
0
def main():
    # TODO: Erase dummy net
    net = None
    """
    TODO: Comment in
    deploy_fpath = 'deploy.prototxt'
    model_fpath = 'model.protobinary'

    caffe.set_mode_gpu()
    net = caffe.Net(deploy_fpath, model_fpath, caffe.TEST)
    """

    bpath = '../yandere-crawler/illust/'

    code_db = []

    # TODO: For train.txt
    with open('./test.txt') as f:
        for line in f:
            preview_fpath, _ = line.rstrip().split(' ')
            preview_fpath = bpath + preview_fpath

            print preview_fpath

            # Make BGR 227x227 image
            data = util.load_image(preview_fpath)

            # Make code
            code = util.make_code(net, data)

            code_db.append((code, preview_fpath))

    with open('./code_db.pickle', 'wb') as f:
        pickle.dump(code_db, f)
Ejemplo n.º 2
0
def main():
    # TODO: Erase dummy net
    net = None
    """
    TODO: Comment in
    deploy_fpath = 'deploy.prototxt'
    model_fpath = 'model.protobinary'

    caffe.set_mode_gpu()
    net = caffe.Net(deploy_fpath, model_fpath, caffe.TEST)
    """

    preview_fpath = '../yandere-crawler/illust/preview/335/335201.jpg'
    code_db_fpath = './code_db.pickle'

    code_db = []
    with open(code_db_fpath, 'rb') as f:
        code_db = pickle.load(f)

    data = util.load_image(preview_fpath)
    code = util.make_code(net, data)

    result = search(code_db, code, 5)
    print result