def horizontal_flip_list()

in lib/datasets/image_processor.py [0:0]

• 16 lines of code
• 8 McCabe index (conditional complexity)

def horizontal_flip_list(prob, images, order='CHW', boxes=None,
                         force_flip=False):

    _, width, _ = images[0].shape
    if np.random.uniform() < prob or force_flip:
        if boxes is not None:
            boxes = flip_boxes(boxes, width)
        if order == 'CHW':
            out_images = []
            for image in images:
                image = np.asarray(image).swapaxes(2, 0)
                image = image[::-1]
                out_images.append(image.swapaxes(0, 2))
            return out_images, boxes
        elif order == 'HWC':
            return [cv2.flip(image, 1) for image in images], boxes
    return images, boxes