CPC G06F 16/58 (2019.01) [G06N 3/04 (2013.01); G06N 3/08 (2013.01)] | 16 Claims |
1. A computer system for automated image retrieval, the computer system comprising:
a processor configured to execute instructions; and
a non-transient computer-readable medium comprising instructions that when executed by the processor cause the processor to:
receive a request from a client device to retrieve images relevant to a query image;
access a set of trained weights for a set of neighbor nodes in an image retrieval graph of a query node associated with the query image, each weight of the set of trained weights representing an edge in the image retrieval graph connecting a respective neighbor node to the query node in the image retrieval graph;
generate a query descriptor mapping the query image to a descriptor space by applying the set of trained weights to combine outputs of the set of neighbor nodes at one or more layers of the image retrieval graph;
identify relevant images based on similarity of image descriptors associated with the relevant images to the query descriptor; and
return information about the relevant images to the client device.
|