mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-15 04:13:57 +08:00
Adios mattport
This commit is contained in:
1 parent
9d2b34ab60
commit
b26408eb7a
86 files changed
+176
-176
No files matched your search
@@ -2,9 +2,9 @@
|
||||
Test the CUDA bindings.
|
||||
"""
|
||||
import torch
|
||||
from mattport.cuda.sampler import Sampler
|
||||
from radiance.cuda.sampler import Sampler
|
||||
import time
|
||||
import mattport_cuda
|
||||
import radiance_cuda
|
||||
import os
|
||||
|
||||
|
||||
@@ -78,7 +78,7 @@ def test_benchmark():
|
||||
print("PYTORCH_SAMPLE_TIME:", PYTORCH_SAMPLE_TIME)
|
||||
|
||||
start = time.time()
|
||||
time_steps_cu, samples_cu, time_steps_mask_cu = mattport_cuda.sample_uniformly_along_ray_bundle(
|
||||
time_steps_cu, samples_cu, time_steps_mask_cu = radiance_cuda.sample_uniformly_along_ray_bundle(
|
||||
origins, directions, nears, fars, offsets, max_num_samples
|
||||
)
|
||||
CUDA_SAMPLE_TIME = time.time() - start
|
||||
|
||||
@@ -7,12 +7,12 @@ import torch
|
||||
from torch.utils.data import DataLoader
|
||||
from tqdm import tqdm
|
||||
|
||||
from mattport.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one
|
||||
from mattport.nerf.dataset.image_dataset import ImageDataset, collate_batch
|
||||
from mattport.nerf.dataset.utils import get_dataset_inputs
|
||||
from mattport.nerf.ray_generator import RayGenerator
|
||||
from mattport.utils.io import get_absolute_path
|
||||
from mattport.viewer.plotly import visualize_dataset
|
||||
from radiance.nerf.dataset.collate import CollateIterDataset, collate_batch_size_one
|
||||
from radiance.nerf.dataset.image_dataset import ImageDataset, collate_batch
|
||||
from radiance.nerf.dataset.utils import get_dataset_inputs
|
||||
from radiance.nerf.ray_generator import RayGenerator
|
||||
from radiance.utils.io import get_absolute_path
|
||||
from radiance.viewer.plotly import visualize_dataset
|
||||
|
||||
|
||||
def visualize_batch(batch):
|
||||
|
||||
@@ -1,7 +1,7 @@
|
||||
"""
|
||||
Embedding tests
|
||||
"""
|
||||
from mattport.nerf.field_modules.embedding import Embedding
|
||||
from radiance.nerf.field_modules.embedding import Embedding
|
||||
|
||||
|
||||
def test_indexing():
|
||||
|
||||
@@ -3,7 +3,7 @@ Encoding Tests
|
||||
"""
|
||||
import pytest
|
||||
import torch
|
||||
from mattport.nerf.field_modules import encoding
|
||||
from radiance.nerf.field_modules import encoding
|
||||
|
||||
|
||||
def test_scaling_and_offset():
|
||||
|
||||
@@ -5,7 +5,7 @@ import torch
|
||||
from torch import nn
|
||||
import pytest
|
||||
|
||||
from mattport.nerf.field_modules.field_heads import (
|
||||
from radiance.nerf.field_modules.field_heads import (
|
||||
DensityFieldHead,
|
||||
FieldHead,
|
||||
FieldHeadNames,
|
||||
|
||||
@@ -4,7 +4,7 @@ MLP Test
|
||||
import torch
|
||||
from torch import nn
|
||||
|
||||
from mattport.nerf.field_modules import MLP
|
||||
from radiance.nerf.field_modules import MLP
|
||||
|
||||
|
||||
def test_mlp():
|
||||
|
||||
@@ -3,9 +3,9 @@ Test the fields
|
||||
"""
|
||||
import torch
|
||||
|
||||
from mattport.nerf.field_modules.field_heads import FieldHeadNames
|
||||
from mattport.nerf.fields.instant_ngp_field import TCNNInstantNGPField
|
||||
from mattport.structures.rays import PointSamples
|
||||
from radiance.nerf.field_modules.field_heads import FieldHeadNames
|
||||
from radiance.nerf.fields.instant_ngp_field import TCNNInstantNGPField
|
||||
from radiance.structures.rays import PointSamples
|
||||
|
||||
|
||||
def test_tcnn_instant_ngp_field():
|
||||
|
||||
@@ -3,7 +3,7 @@ Code to test the occupancy grid.
|
||||
"""
|
||||
|
||||
import torch
|
||||
from mattport.nerf.occupancy_grid import OccupancyGrid
|
||||
from radiance.nerf.occupancy_grid import OccupancyGrid
|
||||
|
||||
|
||||
def test_occupancy_grid():
|
||||
|
||||
@@ -4,7 +4,7 @@ Test renderers
|
||||
import pytest
|
||||
import torch
|
||||
|
||||
from mattport.nerf import renderers
|
||||
from radiance.nerf import renderers
|
||||
|
||||
|
||||
def test_rgb_renderer():
|
||||
|
||||
@@ -2,10 +2,10 @@
|
||||
Test samplers
|
||||
"""
|
||||
import torch
|
||||
from mattport.nerf.sampler import PDFSampler, UniformSampler
|
||||
from radiance.nerf.sampler import PDFSampler, UniformSampler
|
||||
|
||||
from mattport.structures.rays import RayBundle
|
||||
from mattport.nerf.colliders import NearFarCollider
|
||||
from radiance.structures.rays import RayBundle
|
||||
from radiance.nerf.colliders import NearFarCollider
|
||||
|
||||
|
||||
def test_uniform_sampler():
|
||||
|
||||
@@ -4,8 +4,8 @@ Test the camera classes.
|
||||
|
||||
import torch
|
||||
|
||||
from mattport.structures.cameras import PinholeCamera
|
||||
from mattport.structures.rays import RayBundle
|
||||
from radiance.structures.cameras import PinholeCamera
|
||||
from radiance.structures.rays import RayBundle
|
||||
|
||||
|
||||
def test_pinhole_camera():
|
||||
|
||||
Reference in new issue
Block a user