fix: rgb crops batch dimension instead of H/W

[andrewwhitecdw]

Problem

fix: rgb crops batch dimension instead of H/W

Fix

Replace:

    if true_shape is not None:
        H, W = true_shape
        ftensor = ftensor[:H, :W]

with:

    if true_shape is not None:
        H, W = true_shape[-2:]
        if ftensor.ndim == 4:
            ftensor = ftensor[:, :H, :W]
        else:
            ftensor = ftensor[:H, :W]

Files changed

• dust3r/utils/image.py