def augment_data(label_path): label = label_path.split('/')[-1] if label == "0": files = os.listdir(label_path) for file in files: file_name = file.split(".")[0] file_ext = file.split(".")[1] img = Image.open(label_path + "/" + file) img_rotate_120 = img.rotate(120) img_rotate_120.save(label_path + "/" + file_name + "rotate_120." + file_ext) img_enhance = img.filter(ImageFilter.EDGE_ENHANCE()) img_edge = img.filter(ImageFilter.FIND_EDGES()) img_enhance.save(label_path + "/" + file_name + "enhance." + file_ext) img_edge.save(label_path + "/" + file_name + "edge." + file_ext) if label == "1": files = os.listdir(label_path) for file in files: file_name = file.split(".")[0] file_ext = file.split(".")[1] img = Image.open(label_path + "/" + file) img_rotate_120 = img.rotate(120) img_rotate_120.save(label_path + "/" + file_name + "rotate_120." + file_ext) img_enhance = img.filter(ImageFilter.EDGE_ENHANCE()) img_edge = img.filter(ImageFilter.FIND_EDGES()) img_enhance.save(label_path + "/" + file_name + "enhance." + file_ext) img_edge.save(label_path + "/" + file_name + "edge." + file_ext)
def modifyImage(self, some_PIL_image, text='Lorem ipsum'): to_post = some_PIL_image.filter(ImageFilter.GaussianBlur(3)) to_post = some_PIL_image.filter(ImageFilter.SMOOTH_MORE()) to_post = some_PIL_image.filter(ImageFilter.EDGE_ENHANCE()) to_post = self.applyVignette(self.post_path) draw = ImageDraw.Draw(to_post) border_thickness = 4 # offset random_range = [] font_size = 65 if (len(text) < 64): random_range = [(75, 220), (95, 875)] else: random_range = [(50, 220), (65, 475)] randx = random.randint(random_range[0][0], random_range[0][1]) randy = random.randint(random_range[1][0], random_range[1][1]) draw_pos = [randx, randy] reduce = 65 // 7 font_size -= reduce _font = ImageFont.truetype("C:/Windows/fonts/BKANT.ttf", font_size) text = self.formatText(text) draw.text( (draw_pos[0] + border_thickness, draw_pos[1] + border_thickness),\ text, (0,0,0), font = _font)#drop shadow draw.text(draw_pos, text, (255, 255, 255), font=_font) #white text _watermark_font = ImageFont.truetype("C:/Windows/fonts/Arial.ttf", 18) watermark = "@Daily_Philosophy_Trend" draw.text((850, 1050), watermark, (247, 245, 251), font=_watermark_font) to_post.save(self.post_path) return self.post_path
def augment_data(label_path): label = label_path.split('/')[-1] if label == "120": files = os.listdir(label_path) for file in files: file_name = file.split(".")[0] file_ext = file.split(".")[1] img = Image.open(label_path + "/" + file) img_rotate_90 = img.rotate(90) img_rotate_90.save(label_path + "/" + file_name + "rotate_90." + file_ext) if label == "Stop": files = os.listdir(label_path) for file in files: file_name = file.split(".")[0] file_ext = file.split(".")[1] img = Image.open(label_path + "/" + file) img_rotate_90 = img.rotate(90) img_rotate_120 = img.rotate(120) img_rotate_90.save(label_path + "/" + file_name + "rotate_90." + file_ext) img_rotate_120.save(label_path + "/" + file_name + "rotate_120." + file_ext) if label == "Right": files = os.listdir(label_path) for file in files: file_name = file.split(".")[0] file_ext = file.split(".")[1] img = Image.open(label_path + "/" + file) img_enhance = img.filter(ImageFilter.EDGE_ENHANCE()) img_edge = img.filter(ImageFilter.FIND_EDGES()) img_enhance.save(label_path + "/" + file_name + "enhance." + file_ext) img_edge.save(label_path + "/" + file_name + "edge." + file_ext) if label == "Left": files = os.listdir(label_path) for file in files: file_name = file.split(".")[0] file_ext = file.split(".")[1] img = Image.open(label_path + "/" + file) img_enhance = img.filter(ImageFilter.EDGE_ENHANCE()) img_edge = img.filter(ImageFilter.FIND_EDGES()) img_gaussian = img.filter(ImageFilter.GaussianBlur(radius = 3)) img_invert = ImageChops.invert(img) img_enhance.save(label_path + "/" + file_name + "enhance." + file_ext) img_edge.save(label_path + "/" + file_name + "edge." + file_ext) img_gaussian.save(label_path + "/" + file_name + "gaussian." + file_ext) img_invert.save(label_path + "/" + file_name + "invert." + file_ext)
def edge_enhance(image: Image.Image) -> Image.Image: return _apply_filter(image, ImageFilter.EDGE_ENHANCE())