[pose]fix flip_idx (#3577)

single_channel
Laughing 1 year ago committed by GitHub
parent 6a832cdcd3
commit a254087fcd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save