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

Extract finding directories for test data

Browse Source 
Summary: Make common functions for finding directories where test data is found, instead of lots of tests using their own `__file__`  while trying to get ./tests/data and the tutorials data.

Reviewed By: nikhilaravi

Differential Revision: D27633701

fbshipit-source-id: 1467bb6018cea16eba3cab097d713116d51071e9
This commit is contained in:
Rong Rong (AI Infra)
2021-04-08 20:02:16 -07:00
committed by Facebook GitHub Bot
parent 24ee279005
commit 1216b5765a
13 changed files with 81 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
tests/common_testing.py
View File

@@ -9,6 +9,20 @@ import torch
from PIL import Image
def get_tests_dir() -> Path:
"""
Returns Path for the directory containing this file.
"""
return Path(__file__).resolve().parent
def get_pytorch3d_dir() -> Path:
"""
Returns Path for the root PyTorch3D directory.
"""
return get_tests_dir().parent
def load_rgb_image(filename: str, data_dir: Union[str, Path]):
filepath = data_dir / filename
with Image.open(filepath) as raw_image: