INNER CODE UNIT · Python

main

OpenTexture/Paint3D · pipeline_UV_only.py:107

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)

    UV_cnet = txt2imgControlNet(sd_cfg.txt2img)


    total_start = time.time()
    # ===  2. UV gen
    start_t = time.time()
    _ = UV_gen(
        sd_cfg=sd_cfg,

View source record →