image_multithreads.py 文件源码

def image_flip(self, images, labels):
        # ????
        for i in range(len(images)):
            old_image = images[i]
            if numpy.random.random() < 0.5:
                new_image = cv2.flip(old_image, 1)
            else:
                new_image = old_image
            images[i] = new_image

            # ????box label
        for i in range(len(labels)):
            for j in range(len(labels[i])):
                if sum(labels[i][j]) == 0:
                    break
                right = 1.0 - labels[i][j][0]
                left = 1.0 - labels[i][j][1]
                labels[i][j][0] = left
                labels[i][j][1] = right

        return images, labels