423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-22 22:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

Make multisequence evaluator use only cameras within a batch (fixes OOM in v2)

Browse Source 
Summary: In a multisequence (fewview) scenario, it does not make sense to use all cameras for evaluating the difficulty as they come from different scenes. Using only this batch’s source (known) cameras instead.

Reviewed By: bottler

Differential Revision: D38491070

fbshipit-source-id: d6312d8fbb125b28a33db9f53d4215bcd1ca28a8
This commit is contained in:
Roman Shapovalov 2022-08-08 04:43:21 -07:00 committed by Facebook GitHub Bot
parent 0e913b1c2e
commit a393cccf6a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pytorch3d/implicitron/evaluation/evaluator.py
View File

@ -115,7 +115,9 @@ class ImplicitronEvaluator(EvaluatorBase):
implicitron_render,
bg_color="black",
lpips_model=lpips_model,
source_cameras=all_train_cameras,
source_cameras=( # None will make it use batch’s known cameras
None if self.is_multisequence else all_train_cameras
),
)
)