Adding more TODOs

This commit is contained in:
Jake Austin jake-austin@berkeley.edu committed 2023-09-12 03:38:21 +00:00
1 parent bb90b94fb0
commit 8bf21213f1
1 file changed
+2
+2
View File
@@ -331,6 +331,8 @@ class GaussianSplattingModel(Model):
bg_color=torch.ones(3).to(self.device),
)["render"].permute(1, 2, 0) #HWC
# TODO (jake-austin): need to clip the images since there are color overflow issues
return {"rgb": image}
def get_metrics_dict(self, outputs, batch) -> Dict[str, torch.Tensor]: