SortFaceCNN

class tools.sort.sort_methods_aligned.SortFaceCNN(arguments: Namespace, is_group: bool = False)

Bases: SortAlignedMetric

Sort by landmark similarity or dissimilarity

Parameters:
  • arguments (argparse.Namespace) – The command line arguments passed to the sort process

  • is_group (bool, optional) – Set to True if this class is going to be called exclusively for binning. Default: False

Attributes Summary

bin_names

The name of each created bin, if they exist, otherwise an empty list

binned

List of bins (list) containing the filenames belonging to the bin.

loader_type

The loader that this sorter uses

sorted_filelist

List of sorted filenames for given sorter in a single list.

Methods Summary

binning()

Group into bins by CNN face similarity

score_image(filename, image, alignments)

Score a single image for sort method: "distance", "yaw", "pitch" or "size" and add the result to _result

sort()

Sort by landmarks.

Attributes Documentation

bin_names

The name of each created bin, if they exist, otherwise an empty list

binned

List of bins (list) containing the filenames belonging to the bin. The binning process is called when this property is first accessed

loader_type

The loader that this sorter uses

Type:

[“face”, “meta”, “all”]

sorted_filelist

List of sorted filenames for given sorter in a single list. The sort process is called when this property is first accessed

Methods Documentation

binning() list[list[str]]

Group into bins by CNN face similarity

Returns:

List of bins of filenames

Return type:

list

score_image(filename: str, image: np.ndarray | None, alignments: PNGAlignments | None) None

Score a single image for sort method: “distance”, “yaw”, “pitch” or “size” and add the result to _result

Parameters:
  • filename (str) – The filename of the currently processing image

  • image (np.ndarray or None) – A face image loaded from disk or None

  • alignments (dict or None) – The alignments dictionary for the aligned face or None

Return type:

None

sort() None

Sort by landmarks.

Return type:

None

binning() list[list[str]]

Group into bins by CNN face similarity

Returns:

List of bins of filenames

Return type:

list

sort() None

Sort by landmarks.

Return type:

None