Adios mattport

This commit is contained in:
Matt Tancik committed 2022-05-30 09:24:12 -07:00
1 parent 9d2b34ab60
commit b26408eb7a
86 files changed
+176 -176

No files matched your search

+3 -3
View File
@@ -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
+6 -6
View File
@@ -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 -1
View File
@@ -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():
+1 -1
View File
@@ -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,
+1 -1
View File
@@ -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 -3
View File
@@ -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():
+1 -1
View File
@@ -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():
+1 -1
View File
@@ -4,7 +4,7 @@ Test renderers
import pytest
import torch
from mattport.nerf import renderers
from radiance.nerf import renderers
def test_rgb_renderer():
+3 -3
View File
@@ -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():
+2 -2
View File
@@ -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():