fixed error that happens when you try to load LLFF data with no NDC: bds variable is a numpy array but torch.min expects a tensor

This commit is contained in:
Stefan Holzer 2020-04-12 14:14:27 -07:00
parent f913df5946
commit b84f89eec0

View file

@ -445,8 +445,9 @@ def train():
print('DEFINING BOUNDS') print('DEFINING BOUNDS')
if args.no_ndc: if args.no_ndc:
near = torch.min(bds) * .9 near = np.ndarray.min(bds) * .9
far = torch.max(bds) * 1. far = np.ndarray.max(bds) * 1.
else: else:
near = 0. near = 0.
far = 1. far = 1.