In https://github.com/kerrj/lerf/blob/main/lerf/data/lerf_datamanager.py#L100 image size is used as images.shape[2:4] however the shape of images is Nx3xHxW. which makes image size equal to [H, W].
In PatchEmbeddingDataloader center_x is defined using cfg["image_shape"][0] and center_y is defined using cfg["image_shape"][1].
In https://github.com/kerrj/lerf/blob/main/lerf/data/lerf_datamanager.py#L100 image size is used as
images.shape[2:4]
however the shape of images is Nx3xHxW. which makes image size equal to [H, W]. In PatchEmbeddingDataloadercenter_x
is defined usingcfg["image_shape"][0]
andcenter_y
is defined usingcfg["image_shape"][1]
.Am I missing something or is it a bug?