From 97c44e2811b1f177618566935c8b0e060ee8530b Mon Sep 17 00:00:00 2001 From: fe Date: Wed, 3 Jan 2024 12:44:27 -0800 Subject: [PATCH] Update README.md --- README.md | 1 + 1 file changed, 1 insertion(+) diff --git a/README.md b/README.md index e02d4a6..9dd5fd1 100644 --- a/README.md +++ b/README.md @@ -57,6 +57,7 @@ Here is a minimal example assuming you have a pointcloud in the folder data. ```python from mask3d import get_model, load_mesh, prepare_data, map_output_to_pointcloud, save_colorized_mesh +import torch model = get_model('checkpoints/scannet200/scannet200_benchmark.ckpt') model.eval()