INNER CODE UNIT · Python

main

OpenTexture/Paint3D · pipeline_paint3d_stage1.py:185

def main():
    print("Depth-based 3D Texturing")
    opt = parse()
    sd_cfg, render_cfg = init_process(opt)

    # ===  1. create model and data
    device = torch.device("cuda")
    dataloaders = init_dataloaders(render_cfg, device)
    mesh_model = TexturedMeshModel(cfg=render_cfg, device=device,).to(device)

    depth_cnet = txt2imgControlNet(sd_cfg.txt2img)
    inpaint_cnet = inpaintControlNet(sd_cfg.inpaint)


    # ===  2. init view generation
    total_start = time.time()
    start_t = time.time()
    init_images = gen_init_view(

View source record →