SYSTEMS AND METHODS FOR COLOR-BASED OUTFIT CLASSIFICATION
Disclosed herein are systems and method for classifying objects in an image using a color-based machine learning classifier. A method may include: training, with a dataset including a plurality of images, a machine learning classifier to classify an object in a given image into a color class from a set of color classes of a first size; receiving an input image depicting at least one object belonging to the set of color classes; determining a subset of color classes that are anticipated to be in the input image based on metadata of the input image; generating a matched mask input indicating the subset set of color classes in the input image, wherein the subset of color classes is of a second size that is smaller than the first size; and inputting both the input image and the matched mask input into the machine learning classifier.
1 . A method for classifying objects in an image using a color-based machine learning classifier, the method comprising:
training, with a dataset comprising a plurality of images, a machine learning classifier to classify an object in a given image into a color class from a set of color classes each representing a distinct color, wherein the color class represents a predominant color of the object and wherein the set of color classes is of a first size;
receiving an input image depicting at least one object belonging to the set of color classes;
determining, from the set of color classes, a subset of color classes that are anticipated to be in the input image based on metadata of the input image;
generating a matched mask input indicating the subset set of color classes in the input image, wherein the subset of color classes is of a second size that is smaller than the first size;
inputting both the input image and the matched mask input into the machine learning classifier, wherein the machine learning classifier is configured to classify the at least one object into at least one color class of the subset of color classes; and
outputting the at least one color class.
2 . The method of claim 1 , wherein the metadata of the input image comprises a timestamp and an identifier of a source location of the input image, further comprising:
identifying, in a database that maps timestamps to color classes, a list of color classes that are associated with the timestamp of the input image; and
including, in the subset of color classes, color classes in the list.
3 . The method of claim 2 , wherein the database is provided by the source location.
4 . The method of claim 1 , wherein the matched mask input further identifies similar classes that the at least one object does not belong to.
5 . The method of claim 1 , wherein the machine learning classifier is a convolutional neural network.
6 . The method of claim 1 , wherein the machine learning classifier is configured to:
determine, for each respective color class in the set of color classes, a respective probability of the at least one object belonging to the respective color class; and
adjust the respective probability based on whether the respective color class is present in the matched mask input.
7 . The method of claim 1 , wherein the input image is a video frame of a livestream, and wherein the machine learning classifier classifies the at least one object in real-time.
8 . The method of claim 1 , wherein the at least one object is a person wearing an outfit of a particular color.
9 . A system for classifying objects in an image using a color-based machine learning classifier, the system comprising:
a hardware processor configured to:
train, with a dataset comprising a plurality of images, a machine learning classifier to classify an object in a given image into a color class from a set of color classes each representing a distinct color, wherein the color class represents a predominant color of the object and wherein the set of color classes is of a first size;
receive an input image depicting at least one object belonging to the set of color classes;
determine, from the set of color classes, a subset of color classes that are anticipated to be in the input image based on metadata of the input image;
generate a matched mask input indicating the subset set of color classes in the input image, wherein the subset of color classes is of a second size that is smaller than the first size;
input both the input image and the matched mask input into the machine learning classifier, wherein the machine learning classifier is configured to classify the at least one object into at least one color class of the subset of color classes; and
output the at least one color class.
10 . The system of claim 9 , wherein the metadata of the input image comprises a timestamp and an identifier of a source location of the input image, and wherein the hardware processor is further configured to:
identify, in a database that maps timestamps to color classes, a list of color classes that are associated with the timestamp of the input image; and
include, in the subset of color classes, color classes in the list.
11 . The system of claim 10 , wherein the database is provided by the source location.
12 . The system of claim 9 , wherein the matched mask input further identifies similar classes that the at least one object does not belong to.
13 . The system of claim 9 , wherein the machine learning classifier is a convolutional neural network.
14 . The system of claim 9 , wherein the machine learning classifier is configured to:
determine, for each respective color class in the set of color classes, a respective probability of the at least one object belonging to the respective color class; and
adjust the respective probability based on whether the respective color class is present in the matched mask input.
15 . The system of claim 9 , wherein the input image is a video frame of a livestream, and wherein the machine learning classifier classifies the at least one object in real-time.
16 . The system of claim 9 , wherein the at least one object is a person wearing an outfit of a particular color.
17 . A non-transitory computer readable medium storing thereon computer executable instructions for classifying objects in an image using a color-based machine learning classifier, including instructions for:
training, with a dataset comprising a plurality of images, a machine learning classifier to classify an object in a given image into a color class from a set of color classes each representing a distinct color, wherein the color class represents a predominant color of the object and wherein the set of color classes is of a first size;
receiving an input image depicting at least one object belonging to the set of color classes;
determining, from the set of color classes, a subset of color classes that are anticipated to be in the input image based on metadata of the input image;
generating a matched mask input indicating the subset set of color classes in the input image, wherein the subset of color classes is of a second size that is smaller than the first size;
inputting both the input image and the matched mask input into the machine learning classifier, wherein the machine learning classifier is configured to classify the at least one object into at least one color class of the subset of color classes; and
outputting the at least one color class.