synthval.features_extraction.RadDinoFeatureExtractor#
- class synthval.features_extraction.RadDinoFeatureExtractor#
Bases:
FeatureExtractorFeature extractor using the HuggingFace model microsoft/rad-dino for extracting features from images. This implementation supports batch processing for improved performance on large datasets.
- batch_feature_extraction(images)#
Extract features from a batch of images using the HuggingFace Rad-Dino model.
- Parameters:
images (List[PIL.Image.Image]) – List of PIL images to extract features from.
- Returns:
A NumPy array where each row corresponds to the feature vector of an image.
- Return type:
numpy.ndarray