data_juicer.ops.mapper.detect_character_attributes_mapper module

class data_juicer.ops.mapper.detect_character_attributes_mapper.DetectCharacterAttributesMapper(detect_character_locations_mapper_args: Dict | None = {}, *args, **kwargs)

Bases: Mapper

Takes an image, a caption, and main character names as input to extract the characters’ attributes.

__init__(detect_character_locations_mapper_args: Dict | None = {}, *args, **kwargs)

Initialization method.

Parameters:

detect_character_locations_mapper_args – Arguments for detect_character_locations_mapper_args. Controls the threshold for locating the main character. Default empty dict will use fixed values: default mllm_mapper_args, default image_text_matching_filter_args, yoloe_path=”yoloe-11l-seg.pt”, iou_threshold=0.7, matching_score_threshold=0.4,

process_single(samples, rank=None)

For sample level, sample –> sample

Parameters:

sample – sample to process

Returns:

processed sample