diff --git a/data_juicer/ops/filter/face_area_filter.py b/data_juicer/ops/filter/face_area_filter.py index 3819ea0a2..27ce805f2 100644 --- a/data_juicer/ops/filter/face_area_filter.py +++ b/data_juicer/ops/filter/face_area_filter.py @@ -34,8 +34,8 @@ def get_classifier(self): return cv2.CascadeClassifier(cv2.data.haarcascades + self.file_path) -@OPERATORS.register_module('face_area_filter') -@LOADED_IMAGES.register_module('face_area_filter') +@OPERATORS.register_module(OP_NAME) +@LOADED_IMAGES.register_module(OP_NAME) class FaceAreaFilter(Filter): """Filter to keep samples with face area ratio within a specific range. """