SortFaceCNN
- class tools.sort.sort_methods_aligned.SortFaceCNN(arguments: Namespace, is_group: bool = False)
Bases:
SortAlignedMetricSort by landmark similarity or dissimilarity
- Parameters:
arguments (
argparse.Namespace) – The command line arguments passed to the sort processis_group (bool, optional) – Set to
Trueif this class is going to be called exclusively for binning. Default:False
Attributes Summary
The name of each created bin, if they exist, otherwise an empty list
List of bins (list) containing the filenames belonging to the bin.
The loader that this sorter uses
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
_resultsort()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.ndarrayorNone) – A face image loaded from disk orNonealignments (dict or
None) – The alignments dictionary for the aligned face orNone
- 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