INNER CODE UNIT · Python

main

OpenTexture/Paint3D · pipeline_paint3d_stage2.py:148

def main():
    print("Texture diffusion")
    opt = parse()
    sd_cfg, render_cfg = init_process(opt)

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

    UVInpaint_cnet = inpaintControlNet(sd_cfg.inpaint)
    UVtile_cnet = img2imgControlNet(sd_cfg.img2img)

    total_start = time.time()
    # ===  2. UV inpaint
    start_t = time.time()
    UV_inpaint_res = UV_inpaint(
        sd_cfg=sd_cfg,

View source record →