reshape_train_test

cellpose_omni.transforms.reshape_train_test(train_data, train_labels, test_data, test_labels, channels, channel_axis=0, normalize=True, dim=2, omni=False)

check sizes and reshape train and test data for training