SYMBOL INDEX (186 symbols across 24 files)

FILE: add_degradation/add_degradation.py
  function motion_blur (line 6) | def motion_blur(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function lens_blur (line 21) | def lens_blur(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function gaussian_noise (line 43) | def gaussian_noise(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function block_exchange (line 54) | def block_exchange(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function jpeg_compression (line 80) | def jpeg_compression(img: np.ndarray, intensity: float = 0.5) -> np.ndar...
  function mean_shift (line 97) | def mean_shift(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function color_diffusion (line 107) | def color_diffusion(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function sharpness_change (line 137) | def sharpness_change(img: np.ndarray, intensity: float = 0.5) -> np.ndar...
  function dark_illumination (line 152) | def dark_illumination(img: np.ndarray, intensity: float = 0.5) -> np.nda...
  function hsv_saturation (line 160) | def hsv_saturation(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function atmospheric_turbulence (line 171) | def atmospheric_turbulence(img: np.ndarray, intensity: float = 0.5) -> n...
  function dirty_lens (line 184) | def dirty_lens(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function scan_lines (line 223) | def scan_lines(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function graffiti (line 238) | def graffiti(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:
  function watermark_damage (line 279) | def watermark_damage(img: np.ndarray, intensity: float = 0.5) -> np.ndar...
  function lens_flare (line 302) | def lens_flare(img: np.ndarray, intensity: float = 0.5) -> np.ndarray:

FILE: add_degradation/generate_degradation.py
  function apply_degradation_Benchmark (line 34) | def apply_degradation_Benchmark(image, method_name, intensity):
  function main (line 39) | def main():

FILE: app.py
  class ModelHandler (line 28) | class ModelHandler:
    method __init__ (line 29) | def __init__(self, model_path):
    method _load_model (line 35) | def _load_model(self):
    method predict (line 52) | def predict(self, message_dict, history, temperature, max_tokens):
  function create_chat_ui (line 160) | def create_chat_ui():

FILE: demo.py
  class ModelHandler (line 31) | class ModelHandler:
    method __init__ (line 32) | def __init__(self, model_path):
    method _load_model (line 38) | def _load_model(self):
    method predict (line 53) | def predict(self, question, image_path, temperature=DEFAULT_TEMPERATUR...
  function main (line 124) | def main():

FILE: src/eval/test_od_r1.py
  function extract_bbox_answer (line 13) | def extract_bbox_answer(content):
  function iou (line 28) | def iou(box1, box2):
  function load_model (line 41) | def load_model(model_path, device_map):
  function eval_od_r1 (line 56) | def eval_od_r1(

FILE: src/eval/test_rec_baseline.py
  function setup_distributed (line 20) | def setup_distributed():
  function extract_bbox_answer (line 62) | def extract_bbox_answer(content):
  function iou (line 72) | def iou(box1, box2):

FILE: src/eval/test_rec_r1.py
  function setup_distributed (line 20) | def setup_distributed():
  function extract_bbox_answer (line 67) | def extract_bbox_answer(content):
  function iou (line 80) | def iou(box1, box2):

FILE: src/eval/test_rec_r1_internvl.py
  function setup_distributed (line 18) | def setup_distributed():
  function extract_bbox_answer (line 70) | def extract_bbox_answer(content):
  function iou (line 83) | def iou(box1, box2):
  function process_vision_info (line 96) | def process_vision_info(batch_messages):

FILE: src/open-r1-multimodal/setup.py
  function deps_list (line 80) | def deps_list(*pkgs):

FILE: src/open-r1-multimodal/src/open_r1/configs.py
  class GRPOConfig (line 24) | class GRPOConfig(trl.GRPOConfig):
  class SFTConfig (line 54) | class SFTConfig(trl.SFTConfig):

FILE: src/open-r1-multimodal/src/open_r1/evaluate.py
  function prompt_fn (line 37) | def prompt_fn(line, task_name: str = None):

FILE: src/open-r1-multimodal/src/open_r1/generate.py
  function build_distilabel_pipeline (line 22) | def build_distilabel_pipeline(

FILE: src/open-r1-multimodal/src/open_r1/grpo_jsonl.py
  function initialize_tokenizer (line 60) | def initialize_tokenizer(model_path):
  class GRPOScriptArguments (line 68) | class GRPOScriptArguments(ScriptArguments):
  function extract_choice (line 119) | def extract_choice(text):
  function evaluate_answer_similarity (line 165) | def evaluate_answer_similarity(student_answer, ground_truth):
  function llm_reward (line 190) | def llm_reward(content, sol, **kwargs):
  function mcq_reward (line 200) | def mcq_reward(content, sol, **kwargs):
  function yes_no_reward (line 219) | def yes_no_reward(content, sol, **kwargs):
  function calculate_map (line 241) | def calculate_map(pred_bbox_list, gt_bbox_list, score_type=0):
  function map_reward (line 300) | def map_reward(content, sol, length_reward=False, score_type=0, **kwargs):
  function od_reward (line 354) | def od_reward(content, sol, score_type=0, **kwargs):
  function odLength_reward (line 388) | def odLength_reward(content, sol, **kwargs):
  function iou (line 421) | def iou(box1, box2):
  function detection_score (line 433) | def detection_score(content, sol, iou_threshold=0.5, alpha=0.7, beta=0.0...
  function cosine_reward (line 567) | def cosine_reward(content, tokenizer, acc_reward, **kwargs):
  function repetition_reward (line 594) | def repetition_reward(content, **kwargs):
  function repetition_rewards (line 688) | def repetition_rewards(completions, solution, **kwargs):
  function cosine_rewards (line 715) | def cosine_rewards(completions, solution, **kwargs):
  function numeric_reward (line 747) | def numeric_reward(content, sol, **kwargs):
  function math_reward (line 755) | def math_reward(content, sol, **kwargs):
  function clean_text (line 759) | def clean_text(text, exclue_chars=['\n', '\r']):
  function all_match_reward (line 778) | def all_match_reward(content, sol, **kwargs):
  function default_accuracy_reward (line 783) | def default_accuracy_reward(content, sol, **kwargs):
  function accuracy_reward (line 828) | def accuracy_reward(completions, solution, **kwargs):
  function format_reward (line 883) | def format_reward(completions, **kwargs):
  class GRPOModelConfig (line 909) | class GRPOModelConfig(ModelConfig):
  function get_vlm_module (line 920) | def get_vlm_module(model_name_or_path):
  function main (line 926) | def main(script_args, training_args, model_args):

FILE: src/open-r1-multimodal/src/open_r1/qwen2_5vl_monkey_patch.py
  function qwen2_5vl_vision_flash_attn_forward (line 6) | def qwen2_5vl_vision_flash_attn_forward(
  function monkey_patch_qwen2_5vl_flash_attn (line 43) | def monkey_patch_qwen2_5vl_flash_attn():
  function qwen2_5vl_forward (line 52) | def qwen2_5vl_forward(
  function monkey_patch_qwen2_5vl_forward (line 213) | def monkey_patch_qwen2_5vl_forward():
  function weigths_only_load (line 219) | def weigths_only_load(self, path: str, map_location=None):
  function monkey_patch_torch_load (line 225) | def monkey_patch_torch_load():

FILE: src/open-r1-multimodal/src/open_r1/trainer/grpo_config.py
  class GRPOConfig (line 22) | class GRPOConfig(TrainingArguments):

FILE: src/open-r1-multimodal/src/open_r1/trainer/grpo_trainer.py
  class RepeatRandomSampler (line 69) | class RepeatRandomSampler(Sampler):
    method __init__ (line 86) | def __init__(
    method __iter__ (line 104) | def __iter__(self):
    method __len__ (line 115) | def __len__(self) -> int:
  class VLMGRPOTrainer (line 119) | class VLMGRPOTrainer(Trainer):
    method __init__ (line 203) | def __init__(
    method _enable_gradient_checkpointing (line 471) | def _enable_gradient_checkpointing(self, model: PreTrainedModel, args:...
    method _set_signature_columns_if_needed (line 502) | def _set_signature_columns_if_needed(self):
    method _get_per_token_logps (line 512) | def _get_per_token_logps(self, model, input_ids, attention_mask, **cus...
    method _prepare_inputs (line 525) | def _prepare_inputs(self, inputs):
    method _get_key_from_inputs (line 529) | def _get_key_from_inputs(self, x, key):
    method _generate_and_score_completions (line 537) | def _generate_and_score_completions(self, inputs: dict[str, Union[torc...
    method compute_loss (line 720) | def compute_loss(self, model, inputs, return_outputs=False, num_items_...
    method log (line 780) | def log(self, logs: dict[str, float], start_time: Optional[float] = No...
    method create_model_card (line 789) | def create_model_card(
    method _get_train_sampler (line 847) | def _get_train_sampler(self) -> Sampler:
    method _get_eval_sampler (line 863) | def _get_eval_sampler(self, eval_dataset) -> Sampler:

FILE: src/open-r1-multimodal/src/open_r1/utils/callbacks.py
  function is_slurm_available (line 28) | def is_slurm_available() -> bool:
  class DummyConfig (line 37) | class DummyConfig:
    method __init__ (line 38) | def __init__(self, **kwargs):
  class PushToHubRevisionCallback (line 43) | class PushToHubRevisionCallback(TrainerCallback):
    method __init__ (line 44) | def __init__(self, model_config) -> None:
    method on_save (line 47) | def on_save(self, args: TrainingArguments, state: TrainerState, contro...
  function get_callbacks (line 79) | def get_callbacks(train_config, model_config) -> List[TrainerCallback]:

FILE: src/open-r1-multimodal/src/open_r1/utils/evaluation.py
  function register_lighteval_task (line 26) | def register_lighteval_task(
  function get_lighteval_tasks (line 55) | def get_lighteval_tasks():
  function run_lighteval_job (line 62) | def run_lighteval_job(
  function run_benchmark_jobs (line 93) | def run_benchmark_jobs(training_args: Union["SFTConfig", "GRPOConfig"], ...

FILE: src/open-r1-multimodal/src/open_r1/utils/hub.py
  function push_to_hub_revision (line 39) | def push_to_hub_revision(training_args: SFTConfig | GRPOConfig, extra_ig...
  function check_hub_revision_exists (line 70) | def check_hub_revision_exists(training_args: SFTConfig | GRPOConfig):
  function get_param_count_from_repo_id (line 88) | def get_param_count_from_repo_id(repo_id: str) -> int:
  function get_gpu_count_for_vllm (line 120) | def get_gpu_count_for_vllm(model_name: str, revision: str = "main", num_...

FILE: src/open-r1-multimodal/src/open_r1/utils/math.py
  function compute_score (line 2) | def compute_score(solution_str, ground_truth) -> float:
  function remove_boxed (line 25) | def remove_boxed(s):
  function last_boxed_only_string (line 38) | def last_boxed_only_string(string):
  function is_equiv (line 68) | def is_equiv(str1, str2, verbose=False):
  function fix_fracs (line 86) | def fix_fracs(string):
  function fix_a_slash_b (line 118) | def fix_a_slash_b(string):
  function remove_right_units (line 133) | def remove_right_units(string):
  function fix_sqrt (line 143) | def fix_sqrt(string):
  function strip_string (line 158) | def strip_string(string):

FILE: src/open-r1-multimodal/src/open_r1/utils/pycocotools/coco.py
  function _isArrayLike (line 20) | def _isArrayLike(obj):
  class COCO (line 24) | class COCO:
    method __init__ (line 25) | def __init__(self, annotation_file=None):
    method createIndex (line 47) | def createIndex(self):
    method info (line 78) | def info(self):
    method getAnnIds (line 86) | def getAnnIds(self, imgIds=[], catIds=[], areaRng=[], iscrowd=None):
    method getCatIds (line 114) | def getCatIds(self, catNms=[], supNms=[], catIds=[]):
    method getImgIds (line 136) | def getImgIds(self, imgIds=[], catIds=[]):
    method loadAnns (line 157) | def loadAnns(self, ids=[]):
    method loadCats (line 168) | def loadCats(self, ids=[]):
    method loadImgs (line 179) | def loadImgs(self, ids=[]):
    method showAnns (line 190) | def showAnns(self, anns, draw_bbox=False):
    method loadRes (line 262) | def loadRes(self, resFile):
    method download (line 323) | def download(self, tarDir = None, imgIds = [] ):
    method loadNumpyAnnotations (line 347) | def loadNumpyAnnotations(self, data):
    method annToRLE (line 370) | def annToRLE(self, ann):
    method annToMask (line 391) | def annToMask(self, ann):

FILE: src/open-r1-multimodal/src/open_r1/utils/pycocotools/cocoeval.py
  class COCOeval (line 8) | class COCOeval:
    method __init__ (line 58) | def __init__(self, cocoGt=None, cocoDt=None, iouType='segm'):
    method _prepare (line 82) | def _prepare(self):
    method evaluate (line 119) | def evaluate(self):
    method computeIoU (line 161) | def computeIoU(self, imgId, catId):
    method computeOks (line 190) | def computeOks(self, imgId, catId):
    method evaluateImg (line 233) | def evaluateImg(self, imgId, catId, aRng, maxDet):
    method accumulate (line 313) | def accumulate(self, p = None):
    method summarize (line 420) | def summarize(self):
    method __str__ (line 493) | def __str__(self):
  class Params (line 496) | class Params:
    method setDetParams (line 500) | def setDetParams(self):
    method setKpParams (line 511) | def setKpParams(self):
    method __init__ (line 523) | def __init__(self, iouType='segm'):

FILE: src/open-r1-multimodal/src/open_r1/vlm_modules/qwen_module.py
  class Qwen2VLModule (line 11) | class Qwen2VLModule(VLMBaseModule):
    method __init__ (line 12) | def __init__(self):
    method get_vlm_key (line 15) | def get_vlm_key(self):
    method get_model_class (line 18) | def get_model_class(self, model_id: str, model_init_kwargs: dict):
    method post_model_init (line 27) | def post_model_init(self, model, processing_class):
    method get_processing_class (line 30) | def get_processing_class(self):
    method get_vision_modules_keywords (line 33) | def get_vision_modules_keywords(self):
    method get_custom_multimodal_keywords (line 36) | def get_custom_multimodal_keywords(self):
    method get_non_generate_params (line 39) | def get_non_generate_params(self):
    method get_custom_processing_keywords (line 42) | def get_custom_processing_keywords(self):
    method prepare_prompt (line 45) | def prepare_prompt(self, processing_class, inputs: dict[str, Union[tor...
    method prepare_model_inputs (line 49) | def prepare_model_inputs(self, processing_class, prompts_text, images,...
    method get_question_template (line 71) | def get_question_template(task_type: str):
    method format_reward_rec (line 91) | def format_reward_rec(completions, **kwargs):
    method format_reward_robust (line 111) | def format_reward_robust(completions, **kwargs):
    method type_reward (line 131) | def type_reward(completions, solution, **kwargs):
    method accuracy_reward (line 191) | def accuracy_reward(completions, solution, **kwargs):
    method length_reward (line 216) | def length_reward(completions, solution, **kwargs):
    method iou_reward (line 241) | def iou_reward(completions, solution, **kwargs):
    method select_reward_func (line 293) | def select_reward_func(func: str, task_type: str):

FILE: src/open-r1-multimodal/src/open_r1/vlm_modules/vlm_module.py
  class VLMBaseModule (line 6) | class VLMBaseModule(ABC):
    method __init__ (line 7) | def __init__(self):
    method get_vlm_key (line 11) | def get_vlm_key(self):
    method get_model_class (line 15) | def get_model_class(self, model_id: str, model_init_kwargs: dict):
    method post_model_init (line 18) | def post_model_init(self, model, processing_class):
    method is_embeds_input (line 21) | def is_embeds_input(self):
    method get_processing_class (line 25) | def get_processing_class(self):
    method get_vision_modules_keywords (line 29) | def get_vision_modules_keywords(self):
    method get_custom_multimodal_keywords (line 33) | def get_custom_multimodal_keywords(self):
    method get_non_generate_params (line 37) | def get_non_generate_params(self):
    method get_custom_processing_keywords (line 41) | def get_custom_processing_keywords(self):
    method prepare_prompt (line 45) | def prepare_prompt(self, processing_class, inputs: dict[str, Union[tor...
    method prepare_model_inputs (line 49) | def prepare_model_inputs(self, processing_class, prompts_text, images,...