423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2026-01-17 11:50:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding SQL dataset classes to ImplicitronDataSource imports

Browse Source 
Summary: Making it easier for the clients to use these datasets.

Reviewed By: bottler

Differential Revision: D46727179

fbshipit-source-id: cf619aee4c4c0222a74b30ea590cf37f08f014cc
This commit is contained in:
Roman Shapovalov
2023-06-14 10:51:47 -07:00
committed by Facebook GitHub Bot
parent 3d886c32d5
commit cd5db076d5
5 changed files with 41 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -72,6 +72,16 @@ class ImplicitronDataSource(DataSourceBase): # pyre-ignore[13]
from .rendered_mesh_dataset_map_provider import ( # noqa: F401
RenderedMeshDatasetMapProvider,
)
from .train_eval_data_loader_provider import ( # noqa: F401
TrainEvalDataLoaderMapProvider,
)
try:
from .sql_dataset_provider import ( # noqa: F401 # pyre-ignore
SqlIndexDatasetMapProvider,
)
except ModuleNotFoundError:
pass # environment without SQL dataset
finally:
pass