Merge pull request #165 from roboflow/fix-classification

Fix top_k processing in as_folder_structure

supervision/dataset/core.py

@@ -584,7 +584,7 @@ def as_folder_structure(self, root_directory_path: str) -> None:
             class_id = (
                 classification.class_id[0]
                 if classification.confidence is None
-                else classification.get_top_k(1)[0]
+                else classification.get_top_k(1)[0][0]
             )
             class_name = self.classes[class_id]
             image_path = os.path.join(root_directory_path, class_name, image_name)