|
|
|
@ -772,10 +772,10 @@ def v8_transforms(dataset, imgsz, hyp, stretch=False):
|
|
|
|
|
perspective=hyp.perspective,
|
|
|
|
|
pre_transform=None if stretch else LetterBox(new_shape=(imgsz, imgsz)),
|
|
|
|
|
)])
|
|
|
|
|
flip_idx = dataset.data.get('flip_idx', None) # for keypoints augmentation
|
|
|
|
|
flip_idx = dataset.data.get('flip_idx', []) # for keypoints augmentation
|
|
|
|
|
if dataset.use_keypoints:
|
|
|
|
|
kpt_shape = dataset.data.get('kpt_shape', None)
|
|
|
|
|
if flip_idx is None and hyp.fliplr > 0.0:
|
|
|
|
|
if len(flip_idx) == 0 and hyp.fliplr > 0.0:
|
|
|
|
|
hyp.fliplr = 0.0
|
|
|
|
|
LOGGER.warning("WARNING ⚠️ No 'flip_idx' array defined in data.yaml, setting augmentation 'fliplr=0.0'")
|
|
|
|
|
elif flip_idx and (len(flip_idx) != kpt_shape[0]):
|
|
|
|
|