Add new functionality for image rectification and depth metadata handling
- Implemented a preview tool for rectifying images with interactive aspect controls in `preview.py`. - Added tests for depth metadata loading to ensure consistent schema usage in `test_depth_metadata.py`. - Developed comprehensive tests for cutout calibration processes in `test_cutout_calibration.py`, including stability checks and depth calibration. - Created tests for frame compositing operations in `test_frame_compositing.py`, ensuring correct handling of depth and transparency. - Introduced material slot inheritance functionality in `material_slots.py` to manage geometry material assignments. - Built shared shader nodes for material processing in `shader.py`, including roughness and bump mapping setups.
This commit is contained in:
@@ -19,7 +19,7 @@ class ModelInferenceTest(unittest.TestCase):
|
||||
cls.onnx_upscale = load_server_module("models.onnx_upscale")
|
||||
|
||||
|
||||
def test_moge2_batch_releases_only_after_final_image(self):
|
||||
def test_moge2_infers_one_geometry_frame(self):
|
||||
prediction = {
|
||||
"points": np.ones((2, 2, 3), dtype=np.float32),
|
||||
"depth": np.ones((2, 2), dtype=np.float32),
|
||||
@@ -32,18 +32,15 @@ class ModelInferenceTest(unittest.TestCase):
|
||||
"infer_one",
|
||||
return_value=prediction,
|
||||
) as infer_one:
|
||||
frames = self.moge2_model.infer(
|
||||
frame = self.moge2_model.infer(
|
||||
FakeModels(),
|
||||
{"model": "MOGE2_VITS_NORMAL", "device": "cpu"},
|
||||
[Path("a.png"), Path("b.png")],
|
||||
Path("input.png"),
|
||||
include_points=True,
|
||||
)
|
||||
|
||||
self.assertEqual(len(frames), 2)
|
||||
self.assertEqual(
|
||||
[call.kwargs["release_memory"] for call in infer_one.call_args_list],
|
||||
[False, True],
|
||||
)
|
||||
self.assertEqual(frame.depth.shape, (2, 2))
|
||||
infer_one.assert_called_once()
|
||||
|
||||
|
||||
def test_upscale_resource_error_releases_session_and_is_readable(self):
|
||||
|
||||
Reference in New Issue
Block a user