Commit d67662d1 authored by Nikhila Ravi's avatar Nikhila Ravi Committed by Facebook GitHub Bot
Browse files

Update use of `select_cameras`

Summary: Removed `select_cameras.py` from implicitron and updated all callsites to directly index the cameras.

Reviewed By: bottler

Differential Revision: D33187605

fbshipit-source-id: aaf5b36aef9d72db0c7e89dec519f23646f6aa05
parent 28ccdb73
...@@ -383,7 +383,7 @@ class CamerasBase(TensorProperties): ...@@ -383,7 +383,7 @@ class CamerasBase(TensorProperties):
kwargs = {} kwargs = {}
if not isinstance(index, (int, list, torch.LongTensor)): if not isinstance(index, (int, list, torch.LongTensor, torch.cuda.LongTensor)):
msg = "Invalid index type, expected int, List[int] or torch.LongTensor; got %r" msg = "Invalid index type, expected int, List[int] or torch.LongTensor; got %r"
raise ValueError(msg % type(index)) raise ValueError(msg % type(index))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment