我参考'examples/room/condition.npz'的格式生成intrinsics矩阵,(N,3,3),并且只使用了内参 conditions = dict(intrinsics=intrinsics, poses=None, depths=None),使用你提供的模型权重,推理自己的数据集,报错,麻烦看看是什么问题:
Pi3-main\pi3\models\pi3x.py", line 357, in encode
ray_emb = self.ray_embed(rays.reshape(B*N, H, W, 2).permute(0, 3, 1, 2))
RuntimeError: shape '[10, 378, 672, 2]' is invalid for input of size 2032128
我参考'examples/room/condition.npz'的格式生成intrinsics矩阵,(N,3,3),并且只使用了内参 conditions = dict(intrinsics=intrinsics, poses=None, depths=None),使用你提供的模型权重,推理自己的数据集,报错,麻烦看看是什么问题:
Pi3-main\pi3\models\pi3x.py", line 357, in encode
ray_emb = self.ray_embed(rays.reshape(B*N, H, W, 2).permute(0, 3, 1, 2))
RuntimeError: shape '[10, 378, 672, 2]' is invalid for input of size 2032128