PHP 开发中 Elasticsearch 实现图像识别与搜索的技巧

require 'vendor/autoload.php';

$client = ElasticsearchClientBuilder::create()->build();

$params = [
    'index' => 'images',
    'body' => [
        'mappings' => [
            'properties' => [
                'image' => [
                    'type' => 'binary',
                ],
                'tags' => [
                    'type' => 'keyWord',
                ],
            ],
        ],
    ],
];

$response = $client->indices()->create($params);

$imageData = file_get_contents('/path/to/image.jpg');

$params = [
    'index' => 'images',
    'body' => [
        'image' => base64_encode($imageData),
        'tags' => ['sunset', 'beach'],
    ],
];

$response = $client->index($params);

$params = [
    'index' => 'images',
    'body' => [
        'query' => [
            'match' => [
                'tags' => 'sunset',
            ],
        ],
    ],
];

$response = $client->search($params);

require 'vendor/autoload.php';

$graph = new TensorFlowGraph();
$graph->import(new TensorFlowFilesystemLoader('path/to/model.pb'));

$tensor = $graph->createTensorFromPath('path/to/image.jpg');
$session = new TensorFlowSession($graph);
$output = $session->return([$tensor]);
$prediction = $output[0]->data();

$params = [
    'index' => 'images',
    'body' => [
        'query' => [
            'match' => [
                'prediction' => $prediction,
            ],
        ],
    ],
];

$response = $client->search($params);