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

Extract BlobLoader class from JsonIndexDataset and moving crop_by_bbox to FrameData

Browse Source 
Summary:
extracted blob loader
added documentation for blob_loader
did some refactoring on fields
for detailed steps and discussions see:
https://github.com/facebookresearch/pytorch3d/pull/1463
https://github.com/fairinternal/pixar_replay/pull/160

Reviewed By: bottler

Differential Revision: D44061728

fbshipit-source-id: eefb21e9679003045d73729f96e6a93a1d4d2d51
This commit is contained in:
Ildar Salakhiev
2023-04-04 07:17:43 -07:00
committed by Facebook GitHub Bot
parent c759fc560f
commit ebdbfde0ce
15 changed files with 1421 additions and 694 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/implicitron/test_evaluation.py
View File

@@ -13,8 +13,10 @@ import os
import unittest
import lpips
import numpy as np
import torch
from pytorch3d.implicitron.dataset.dataset_base import FrameData
from pytorch3d.implicitron.dataset.frame_data import FrameData
from pytorch3d.implicitron.dataset.json_index_dataset import JsonIndexDataset
from pytorch3d.implicitron.evaluation.evaluate_new_view_synthesis import eval_batch
from pytorch3d.implicitron.models.base_model import ImplicitronModelBase
@@ -268,7 +270,7 @@ class TestEvaluation(unittest.TestCase):
for metric in lower_better:
m_better = eval_result[metric]
m_worse = eval_result_bad[metric]
if m_better != m_better or m_worse != m_worse:
if np.isnan(m_better) or np.isnan(m_worse):
continue # metric is missing, i.e. NaN
_assert = (
self.assertLessEqual