Function dump_as_core_artifacts

dali/test/python/test_utils.py:215–248

(image_info, lhs, rhs, iter=None, sample_idx=None)

def dump_as_core_artifacts(image_info, lhs, rhs, iter=None, sample_idx=None):
    import_numpy()
    import_pil()

    from pathlib import Path

    path = (
        "/opt/dali"
        if os.path.exists("/opt/dali") and os.access("/opt/dali", os.W_OK)
        else os.getcwd()
    )
    Path(f"{path}/core_artifacts").mkdir(parents=True, exist_ok=True)

    image_info = image_info.replace("/", "_")
    image_info = image_info.replace(" ", "_")
    if iter is not None:
        image_info = image_info + f"_iter{iter}"
    if sample_idx is not None:
        image_info = image_info + f"_sample_idx{sample_idx}"

    try:
        save_image(lhs, f"{path}/core_artifacts/{image_info}.lhs.png")
        save_image(rhs, f"{path}/core_artifacts/{image_info}.rhs.png")
    except Exception as e:
        print(f"Tried to save images but got an error: {e}")

    try:
        # save arrays on artifact folder
        import numpy as np

        np.save(f"{path}/core_artifacts/{image_info}.lhs.npy", lhs)
        np.save(f"{path}/core_artifacts/{image_info}.rhs.npy", rhs)
    except Exception as e:
        print(f"Tried to save arrays but got an error: {e}")


# If the `max_allowed_error` is not None, it's checked instead of comparing mean error with `eps`.