Function prepare_gt_batch

train_rgb2material.py:143–150

(batch, hyperparams, do_augmentation=False)

def prepare_gt_batch(batch, hyperparams, do_augmentation=False):
    gt_dict = {}

    
    gt_dict["dinov2_latents"]=batch["latents"]["dinov2"]["final_latent"].cuda()
    gt_dict["material"]=batch["material"].cuda()

    return gt_dict

#