Fix get_pixel_grid when only camera is provided

Phil26AT · Phil26AT · commit c47f9d51c100 · 2025-09-16T10:18:00.000+02:00
diff --git a/gluefactory/utils/misc.py b/gluefactory/utils/misc.py
@@ -444,7 +444,7 @@ def get_pixel_grid(
         ),
         dim=-1,
     )
-    if fmap.ndim == 4:
+    if fmap is not None and fmap.ndim == 4:
         b = fmap.shape[0]
         grid = grid[None].expand(b, -1, -1, -1)
     grid += 0.5

Original file line number	Diff line number	Diff line change
`@@ -444,7 +444,7 @@ def get_pixel_grid(`
`444`	`444`	`),`
`445`	`445`	`dim=-1,`
`446`	`446`	`)`
`447`		`- if fmap.ndim == 4:`
	`447`	`+ if fmap is not None and fmap.ndim == 4:`
`448`	`448`	`b = fmap.shape[0]`
`449`	`449`	`grid = grid[None].expand(b, -1, -1, -1)`
`450`	`450`	`grid += 0.5`