fix ModelNet10 download

LarsKue · Jan 15, 2024 · 9b0881c · 9b0881c
1 parent 8b570b7
commit 9b0881c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/datasets/modelnet10.py b/src/datasets/modelnet10.py
@@ -91,7 +91,7 @@ def download(self):
 
         print(f"Cleaning up hierarchy...")
         # structure is now root/ModelNet10/..., remove intermediate ModelNet10 folder
-        for path in target.glob("archive/*"):
+        for path in target.glob("ModelNet10/*"):
             shutil.move(str(path), str(self.root))
 
         shutil.rmtree(str(self.root / "__MACOSX"))