fix difference of high version albumentations

mindyolo/data/albumentations.py

@@ -56,7 +56,7 @@ def __call__(self, sample, p=1.0, **kwargs):
 
             sample['img'] = new['image']
             sample['bboxes'] = np.array(new['bboxes'])
-            sample['cls'] = np.array(new['class_labels'])
+            sample['cls'] = np.array(new['class_labels']).reshape(-1, 1)
             sample['bbox_format'] = "xywhn"
 
         return sample