def main()

in image_similarity/sanity_checks.py [0:0]

• 56 lines of code
• 2 McCabe index (conditional complexity)

def main(cfg: DictConfig):
    image_net_dir = get_image_net_dir()
    class_labels = get_class_labels(cfg, image_net_dir)

    try:
        env = submitit.JobEnvironment()
        log.info(f"job id: {env.job_id}")
    except RuntimeError:
        warnings.warn("Not launched with submitit")

    across_v_within_similarity = run_within_v_across(
        image_net_dir,
        class_labels,
        cfg.sanity_checks.similarity_type,
        max_n_image_pairs=cfg.sanity_checks.max_n_image_pairs,
        max_class_label_combinations=cfg.sanity_checks.max_class_label_combinations,
        randomly_rotate=False,
        pairs_transform=None,
    )
    log.info(f"across vs. within similairty: {across_v_within_similarity}")

    transformed_similarity = run_transformed_similarities(
        image_net_dir,
        class_labels,
        cfg.sanity_checks.similarity_type,
        max_n_image_pairs=cfg.sanity_checks.max_n_image_pairs,
        small_angle=cfg.sanity_checks.small_angle,
        large_angle=cfg.sanity_checks.large_angle,
    )
    log.info(f"small versus large rotation similarity: {transformed_similarity}")

    across_indep_transformed_similarity = run_within_v_across_indep_transformed(
        image_net_dir,
        class_labels,
        cfg.sanity_checks.similarity_type,
        max_n_image_pairs=cfg.sanity_checks.max_n_image_pairs,
        max_class_label_combinations=cfg.sanity_checks.max_class_label_combinations,
    )
    log.info(
        f"across class independently transformed similarity: "
        f"{across_indep_transformed_similarity}"
    )

    across_pairwise_transformed_similarity = run_within_v_across_pairwise_transformed(
        image_net_dir,
        class_labels,
        cfg.sanity_checks.similarity_type,
        max_n_image_pairs=cfg.sanity_checks.max_n_image_pairs,
        max_class_label_combinations=cfg.sanity_checks.max_class_label_combinations,
    )
    log.info(
        f"across class pairwise transformed similarity: "
        f"{across_pairwise_transformed_similarity}"
    )

    summary = summarize_results(
        across_v_within_similarity,
        transformed_similarity,
        across_indep_transformed_similarity,
        across_pairwise_transformed_similarity,
    )
    log.info(f"summary for {cfg.sanity_checks.similarity_type} \n {summary}")