-
Notifications
You must be signed in to change notification settings - Fork 743
the ply result of demo_multi_object has no color #178
Description
the ply result of demo_multi_object has no color:
code is:
import os
import uuid
import imageio
import numpy as np
from IPython.display import Image as ImageDisplay
import sys
sys.path.append("notebook")
from inference import Inference, ready_gaussian_for_video_rendering, load_image, load_masks, display_image, make_scene, render_video, interactive_visualizer
PATH = os.getcwd()
TAG = "hf"
config_path = "/workspace/sam-3d-objects/checkpoints/pipeline.yaml"
inference = Inference(config_path, compile=False)
IMAGE_PATH = "/workspace/sam-3d-objects/notebook/images/shutterstock_stylish_kidsroom_1640806567/image.png"
IMAGE_NAME = os.path.basename(os.path.dirname(IMAGE_PATH))
image = load_image(IMAGE_PATH)
masks = load_masks(os.path.dirname(IMAGE_PATH), extension=".png")
outputs = [inference(image, mask, seed=42) for mask in masks]
scene_gs = make_scene(*outputs)
scene_gs.save_ply("merged_posed.ply")
result is:
