Function main

inference_recon.py:27–43

()

parser.add_argument("--save_path",type=str, default="results/arkitscenes_results.ply")

def main():
    args = parser.parse_args()
    device = "cuda" if torch.cuda.is_available() else "cpu"
    print(f"Using device: {device}")

    image_folder = args.image_folder
    image_names = [os.path.join(image_folder, img_name) for img_name in os.listdir(image_folder)]
    print(image_names)
  
    model, tokenizer, new_token_ids , vit_image_transform, dino_transform = load_model_and_tokenizer(args)
    pred = model.recon(
        tokenizer,
        new_token_ids,
        dino_transform,
        image_names, 
    )
    save_ply_visualization(pred, args.save_path)

if __name__ == "__main__":
    main()