now offering "normals" in sample dict

2019-08-02 10:49:05 +02:00
parent 0f28969ec7
commit 9c100b6c43
6 changed files with 39 additions and 36 deletions
--- a/predict/data/processed/predict.pt
+++ b/predict/data/processed/predict.pt
--- a/predict/predict.py
+++ b/predict/predict.py
@@ -28,7 +28,8 @@ if __name__ == '__main__':
        root_dir=opt.dataset,
        num_classes=4,
        transform=None,
-        npoints=opt.npoints
+        npoints=opt.npoints,
+        refresh=True
    )
    num_classes = test_dataset.num_classes()