AssertionError: Output shape does not match input shape. Data loss has occured.

[R-N]

Code:

dataset = importer.ImportYoloV5("labels", path_to_images="../images")
dataset.splitter.StratifiedGroupShuffleSplit(train_pct=.8, val_pct=.0, test_pct=.2, batch_size=1)

Environment:

• Windows 11 64 bit
• Python 3.10
• Jupyter notebook

Error:

--------------------------------------------------------------------------- AssertionError Traceback (most recent call last) Cell In[13], line 1 ----> 1 dataset.splitter.StratifiedGroupShuffleSplit(train_pct=.8, val_pct=.0, test_pct=.2, batch_size=1) File ~\AppData\Roaming\Python\Python310\site-packages\pylabel\splitter.py:223, in Split.StratifiedGroupShuffleSplit(self, train_pct, test_pct, val_pct, weight, group_col, cat_col, batch_size) 218 df_val["split"] = "val" 220 df = pd.concat([df_train, pd.concat([df_test, df_val])]) 222 assert ( --> 223 df.shape == df_main.shape 224 ), "Output shape does not match input shape. Data loss has occured." 226 self.dataset.df = df 227 self.dataset.df = self.dataset.df.reset_index(drop=True) AssertionError: Output shape does not match input shape. Data loss has occured.

I have no idea what that means and what I should (or shouldn't) do.

[alexheat]

@R-N I have resolved the issue in the latest version.

[R-N]

@alexheat Thank you. The error went away, but ShowClassSplits is empty.

[R-N]

I see the problem. dataset.df["cat_name"] is empty

[R-N]

Okay it seems to be empty from the start, right after loading, so it's a different issue.