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

register ImplicitronDataSource

Browse Source 
Summary: Just register ImplicitronDataSource. We don't use it as pluggable yet here.

Reviewed By: shapovalov

Differential Revision: D37315698

fbshipit-source-id: ac41153383f9ab6b14ac69a3dfdc44aca0d94995
This commit is contained in:
Jeremy Reizenstein 2022-06-22 04:24:14 -07:00 committed by Facebook GitHub Bot
parent 81d63c6382
commit 2e42ef793f
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pytorch3d/implicitron/dataset/data_source.py
View File

@ -6,7 +6,11 @@
from typing import Tuple
from pytorch3d.implicitron.tools.config import ReplaceableBase, run_auto_creation
from pytorch3d.implicitron.tools.config import (
registry,
ReplaceableBase,
run_auto_creation,
)
from .blender_dataset_map_provider import BlenderDatasetMapProvider # noqa
from .data_loader_map_provider import DataLoaderMap, DataLoaderMapProviderBase
@ -25,6 +29,7 @@ class DataSourceBase(ReplaceableBase):
raise NotImplementedError()
@registry.register
class ImplicitronDataSource(DataSourceBase): # pyre-ignore[13]
"""
Represents the data used in Implicitron. This is the only implementation