├── requirements.txt ├── img └── car.jpg ├── output └── 1.jpg ├── video └── test.mp4 ├── screenshot └── 0.jpg ├── README.md ├── LICENSE ├── .gitignore ├── ascii_art.py ├── pixel_art.py └── ascii_art_color.py /requirements.txt: -------------------------------------------------------------------------------- 1 | pygame 2 | opencv-python 3 | numba 4 | numpy 5 | 6 | -------------------------------------------------------------------------------- /img/car.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StanislavPetrovV/ASCII-art-PIXEL-art/HEAD/img/car.jpg -------------------------------------------------------------------------------- /output/1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StanislavPetrovV/ASCII-art-PIXEL-art/HEAD/output/1.jpg -------------------------------------------------------------------------------- /video/test.mp4: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StanislavPetrovV/ASCII-art-PIXEL-art/HEAD/video/test.mp4 -------------------------------------------------------------------------------- /screenshot/0.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/StanislavPetrovV/ASCII-art-PIXEL-art/HEAD/screenshot/0.jpg -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- 1 | # ASCII-art-PIXEL-art 2 | ASCII art and PIXEL art in Python [ Pygame + OpenCV ] 3 | 4 | Convert video or photo to asсii art or pixel art 5 | 6 | press the "s" key - take a screenshot 7 | 8 | press "r" key - record video (ESC - stop record) 9 | 10 | 11 | ![ascii_pixel_art](screenshot/0.jpg "ascii_pixel_art") 12 | -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- 1 | MIT License 2 | 3 | Copyright (c) 2020 StanislavPetrovV 4 | 5 | Permission is hereby granted, free of charge, to any person obtaining a copy 6 | of this software and associated documentation files (the "Software"), to deal 7 | in the Software without restriction, including without limitation the rights 8 | to use, copy, modify, merge, publish, distribute, sublicense, and/or sell 9 | copies of the Software, and to permit persons to whom the Software is 10 | furnished to do so, subject to the following conditions: 11 | 12 | The above copyright notice and this permission notice shall be included in all 13 | copies or substantial portions of the Software. 14 | 15 | THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 16 | IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 17 | FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE 18 | AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER 19 | LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, 20 | OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE 21 | SOFTWARE. 22 | -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- 1 | # Byte-compiled / optimized / DLL files 2 | __pycache__/ 3 | *.py[cod] 4 | *$py.class 5 | 6 | # C extensions 7 | *.so 8 | 9 | # Distribution / packaging 10 | .Python 11 | build/ 12 | develop-eggs/ 13 | dist/ 14 | downloads/ 15 | eggs/ 16 | .eggs/ 17 | lib/ 18 | lib64/ 19 | parts/ 20 | sdist/ 21 | var/ 22 | wheels/ 23 | pip-wheel-metadata/ 24 | share/python-wheels/ 25 | *.egg-info/ 26 | .installed.cfg 27 | *.egg 28 | MANIFEST 29 | 30 | # PyInstaller 31 | # Usually these files are written by a python script from a template 32 | # before PyInstaller builds the exe, so as to inject date/other infos into it. 33 | *.manifest 34 | *.spec 35 | 36 | # Installer logs 37 | pip-log.txt 38 | pip-delete-this-directory.txt 39 | 40 | # Unit test / coverage reports 41 | htmlcov/ 42 | .tox/ 43 | .nox/ 44 | .coverage 45 | .coverage.* 46 | .cache 47 | nosetests.xml 48 | coverage.xml 49 | *.cover 50 | *.py,cover 51 | .hypothesis/ 52 | .pytest_cache/ 53 | 54 | # Translations 55 | *.mo 56 | *.pot 57 | 58 | # Django stuff: 59 | *.log 60 | local_settings.py 61 | db.sqlite3 62 | db.sqlite3-journal 63 | 64 | # Flask stuff: 65 | instance/ 66 | .webassets-cache 67 | 68 | # Scrapy stuff: 69 | .scrapy 70 | 71 | # Sphinx documentation 72 | docs/_build/ 73 | 74 | # PyBuilder 75 | target/ 76 | 77 | # Jupyter Notebook 78 | .ipynb_checkpoints 79 | 80 | # IPython 81 | profile_default/ 82 | ipython_config.py 83 | 84 | # pyenv 85 | .python-version 86 | 87 | # pipenv 88 | # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control. 89 | # However, in case of collaboration, if having platform-specific dependencies or dependencies 90 | # having no cross-platform support, pipenv may install dependencies that don't work, or not 91 | # install all needed dependencies. 92 | #Pipfile.lock 93 | 94 | # PEP 582; used by e.g. github.com/David-OConnor/pyflow 95 | __pypackages__/ 96 | 97 | # Celery stuff 98 | celerybeat-schedule 99 | celerybeat.pid 100 | 101 | # SageMath parsed files 102 | *.sage.py 103 | 104 | # Environments 105 | .env 106 | .venv 107 | env/ 108 | venv/ 109 | ENV/ 110 | env.bak/ 111 | venv.bak/ 112 | 113 | # Spyder project settings 114 | .spyderproject 115 | .spyproject 116 | 117 | # Rope project settings 118 | .ropeproject 119 | 120 | # mkdocs documentation 121 | /site 122 | 123 | # mypy 124 | .mypy_cache/ 125 | .dmypy.json 126 | dmypy.json 127 | 128 | # Pyre type checker 129 | .pyre/ 130 | -------------------------------------------------------------------------------- /ascii_art.py: -------------------------------------------------------------------------------- 1 | import pygame as pg 2 | from numba import njit 3 | import cv2 4 | 5 | 6 | @njit(fastmath=True) 7 | def accelerate_conversion(image, width, height, ascii_coeff, step): 8 | array_of_values = [] 9 | for x in range(0, width, step): 10 | for y in range(0, height, step): 11 | char_index = image[x, y] // ascii_coeff 12 | if char_index: 13 | array_of_values.append((char_index, (x, y))) 14 | return array_of_values 15 | 16 | 17 | class ArtConverter: 18 | def __init__(self, path='video/test.mp4', font_size=12): 19 | pg.init() 20 | self.path = path 21 | self.capture = cv2.VideoCapture(path) 22 | self.image = self.get_image() 23 | self.RES = self.WIDTH, self.HEIGHT = self.image.shape[0], self.image.shape[1] 24 | self.surface = pg.display.set_mode(self.RES) 25 | self.clock = pg.time.Clock() 26 | 27 | self.ASCII_CHARS = ' .",:;!~+-xmo*#W&8@' 28 | self.ASCII_COEFF = 255 // (len(self.ASCII_CHARS) - 1) 29 | 30 | self.font = pg.font.SysFont('Сourier', font_size, bold=True) 31 | self.CHAR_STEP = int(font_size * 0.6) 32 | self.RENDERED_ASCII_CHARS = [self.font.render(char, False, 'white') for char in self.ASCII_CHARS] 33 | 34 | self.rec_fps = self.capture.get(cv2.CAP_PROP_FPS) 35 | self.record = False 36 | self.recorder = cv2.VideoWriter('output/ascii.mp4', cv2.VideoWriter_fourcc(*'mp4v'), self.rec_fps, self.RES) 37 | 38 | def get_frame(self): 39 | frame = pg.surfarray.array3d(self.surface) 40 | return cv2.transpose(frame) 41 | 42 | def record_frame(self): 43 | if self.record: 44 | frame = self.get_frame() 45 | self.recorder.write(frame) 46 | cv2.imshow('Frame', frame) 47 | if cv2.waitKey(1) & 0xFF == 27: 48 | self.record = not self.record 49 | cv2.destroyAllWindows() 50 | 51 | def draw_converted_image(self): 52 | self.image = self.get_image() 53 | array_of_values = accelerate_conversion(self.image, self.WIDTH, self.HEIGHT, self.ASCII_COEFF, self.CHAR_STEP) 54 | for char_index, pos in array_of_values: 55 | self.surface.blit(self.RENDERED_ASCII_CHARS[char_index], pos) 56 | 57 | def get_image(self): 58 | # self.cv2_image = cv2.imread(self.path) 59 | ret, self.cv2_image = self.capture.read() 60 | if not ret: 61 | exit() 62 | transposed_image = cv2.transpose(self.cv2_image) 63 | gray_image = cv2.cvtColor(transposed_image, cv2.COLOR_BGR2GRAY) 64 | return gray_image 65 | 66 | def draw_cv2_image(self): 67 | resized_cv2_image = cv2.resize(self.cv2_image, (640, 360), interpolation=cv2.INTER_AREA) 68 | cv2.imshow('img', resized_cv2_image) 69 | 70 | def draw(self): 71 | self.surface.fill('black') 72 | self.draw_converted_image() 73 | # self.draw_cv2_image() 74 | 75 | def save_image(self): 76 | pygame_image = pg.surfarray.array3d(self.surface) 77 | cv2_img = cv2.transpose(pygame_image) 78 | cv2.imwrite('output/ascii_image.jpg', cv2_img) 79 | 80 | def run(self): 81 | while True: 82 | for i in pg.event.get(): 83 | if i.type == pg.QUIT: 84 | exit() 85 | elif i.type == pg.KEYDOWN: 86 | if i.key == pg.K_s: 87 | self.save_image() 88 | if i.key == pg.K_r: 89 | self.record = not self.record 90 | self.record_frame() 91 | self.draw() 92 | pg.display.set_caption(str(self.clock.get_fps())) 93 | pg.display.flip() 94 | self.clock.tick() 95 | 96 | 97 | if __name__ == '__main__': 98 | app = ArtConverter() 99 | app.run() -------------------------------------------------------------------------------- /pixel_art.py: -------------------------------------------------------------------------------- 1 | import pygame as pg 2 | import numpy as np 3 | from numba import njit 4 | import pygame.gfxdraw 5 | import cv2 6 | 7 | 8 | @njit(fastmath=True) 9 | def accelerate_conversion(image, width, height, color_coeff, step): 10 | array_of_values = [] 11 | for x in range(0, width, step): 12 | for y in range(0, height, step): 13 | r, g, b = image[x, y] // color_coeff 14 | if r + g + b: 15 | array_of_values.append(((r, g, b), (x, y))) 16 | return array_of_values 17 | 18 | 19 | class ArtConverter: 20 | def __init__(self, path='video/test.mp4', pixel_size=7, color_lvl=8): 21 | pg.init() 22 | self.path = path 23 | self.capture = cv2.VideoCapture(path) 24 | self.PIXEL_SIZE = pixel_size 25 | self.COLOR_LVL = color_lvl 26 | self.image = self.get_image() 27 | self.RES = self.WIDTH, self.HEIGHT = self.image.shape[0], self.image.shape[1] 28 | self.surface = pg.display.set_mode(self.RES) 29 | self.clock = pg.time.Clock() 30 | self.PALETTE, self.COLOR_COEFF = self.create_palette() 31 | 32 | self.rec_fps = self.capture.get(cv2.CAP_PROP_FPS) 33 | self.record = False 34 | self.recorder = cv2.VideoWriter('output/pixel_art.mp4', cv2.VideoWriter_fourcc(*'mp4v'), self.rec_fps, self.RES) 35 | 36 | def get_frame(self): 37 | frame = pg.surfarray.array3d(self.surface) 38 | frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR) 39 | return cv2.transpose(frame) 40 | 41 | def record_frame(self): 42 | if self.record: 43 | frame = self.get_frame() 44 | self.recorder.write(frame) 45 | cv2.imshow('Frame', frame) 46 | if cv2.waitKey(1) & 0xFF == 27: 47 | self.record = not self.record 48 | cv2.destroyAllWindows() 49 | 50 | def draw_converted_image(self): 51 | self.image = self.get_image() 52 | array_of_values = accelerate_conversion(self.image, self.WIDTH, self.HEIGHT, self.COLOR_COEFF, self.PIXEL_SIZE) 53 | for color_key, (x, y) in array_of_values: 54 | color = self.PALETTE[color_key] 55 | pygame.gfxdraw.box(self.surface, (x, y, self.PIXEL_SIZE, self.PIXEL_SIZE), color) 56 | 57 | def create_palette(self): 58 | colors, color_coeff = np.linspace(0, 255, num=self.COLOR_LVL, dtype=int, retstep=True) 59 | color_palette = [np.array([r, g, b]) for r in colors for g in colors for b in colors] 60 | palette = {} 61 | color_coeff = int(color_coeff) 62 | for color in color_palette: 63 | color_key = tuple(color // color_coeff) 64 | palette[color_key] = color 65 | return palette, color_coeff 66 | 67 | def get_image(self): 68 | ret, self.cv2_image = self.capture.read() 69 | if not ret: 70 | exit() 71 | transposed_image = cv2.transpose(self.cv2_image) 72 | image = cv2.cvtColor(transposed_image, cv2.COLOR_BGR2RGB) 73 | return image 74 | 75 | def draw_cv2_image(self): 76 | resized_cv2_image = cv2.resize(self.cv2_image, (640, 360), interpolation=cv2.INTER_AREA) 77 | cv2.imshow('img', resized_cv2_image) 78 | 79 | def draw(self): 80 | self.surface.fill('black') 81 | self.draw_converted_image() 82 | # self.draw_cv2_image() 83 | 84 | def save_image(self): 85 | pygame_image = pg.surfarray.array3d(self.surface) 86 | cv2_img = cv2.transpose(pygame_image) 87 | cv2_img = cv2.cvtColor(cv2_img, cv2.COLOR_RGB2BGR) 88 | cv2.imwrite('output/pixel_art_image.jpg', cv2_img) 89 | 90 | def run(self): 91 | while True: 92 | for i in pg.event.get(): 93 | if i.type == pg.QUIT: 94 | exit() 95 | elif i.type == pg.KEYDOWN: 96 | if i.key == pg.K_s: 97 | self.save_image() 98 | if i.key == pg.K_r: 99 | self.record = not self.record 100 | self.record_frame() 101 | self.draw() 102 | pg.display.set_caption(str(self.clock.get_fps())) 103 | pg.display.flip() 104 | self.clock.tick() 105 | 106 | 107 | if __name__ == '__main__': 108 | app = ArtConverter() 109 | app.run() -------------------------------------------------------------------------------- /ascii_art_color.py: -------------------------------------------------------------------------------- 1 | import pygame as pg 2 | import numpy as np 3 | from numba import njit 4 | import cv2 5 | 6 | 7 | @njit(fastmath=True) 8 | def accelerate_conversion(image, gray_image, width, height, color_coeff, ascii_coeff, step): 9 | array_of_values = [] 10 | for x in range(0, width, step): 11 | for y in range(0, height, step): 12 | char_index = gray_image[x, y] // ascii_coeff 13 | if char_index: 14 | r, g, b = image[x, y] // color_coeff 15 | array_of_values.append((char_index, (r, g, b), (x, y))) 16 | return array_of_values 17 | 18 | 19 | class ArtConverter: 20 | def __init__(self, path='video/test.mp4', font_size=12, color_lvl=8): 21 | pg.init() 22 | self.path = path 23 | self.capture = cv2.VideoCapture(path) 24 | self.COLOR_LVL = color_lvl 25 | self.image, self.gray_image = self.get_image() 26 | self.RES = self.WIDTH, self.HEIGHT = self.image.shape[0], self.image.shape[1] 27 | self.surface = pg.display.set_mode(self.RES) 28 | self.clock = pg.time.Clock() 29 | 30 | self.ASCII_CHARS = ' ixzao*#MW&8%B@$' 31 | self.ASCII_COEFF = 255 // (len(self.ASCII_CHARS) - 1) 32 | 33 | self.font = pg.font.SysFont('Сourier', font_size, bold=True) 34 | self.CHAR_STEP = int(font_size * 0.6) 35 | self.PALETTE, self.COLOR_COEFF = self.create_palette() 36 | 37 | self.rec_fps = self.capture.get(cv2.CAP_PROP_FPS) 38 | self.record = False 39 | self.recorder = cv2.VideoWriter('output/ascii_col.mp4', cv2.VideoWriter_fourcc(*'mp4v'), self.rec_fps, self.RES) 40 | 41 | def get_frame(self): 42 | frame = pg.surfarray.array3d(self.surface) 43 | frame = cv2.cvtColor(frame, cv2.COLOR_RGB2BGR) 44 | return cv2.transpose(frame) 45 | 46 | def record_frame(self): 47 | if self.record: 48 | frame = self.get_frame() 49 | self.recorder.write(frame) 50 | cv2.imshow('Frame', frame) 51 | if cv2.waitKey(1) & 0xFF == 27: 52 | self.record = not self.record 53 | cv2.destroyAllWindows() 54 | 55 | def draw_converted_image(self): 56 | image, gray_image = self.get_image() 57 | array_of_values = accelerate_conversion(image, gray_image, self.WIDTH, self.HEIGHT, 58 | self.COLOR_COEFF, self.ASCII_COEFF, self.CHAR_STEP) 59 | for char_index, color, pos in array_of_values: 60 | char = self.ASCII_CHARS[char_index] 61 | self.surface.blit(self.PALETTE[char][color], pos) 62 | 63 | def create_palette(self): 64 | colors, color_coeff = np.linspace(0, 255, num=self.COLOR_LVL, dtype=int, retstep=True) 65 | color_palette = [np.array([r, g, b]) for r in colors for g in colors for b in colors] 66 | palette = dict.fromkeys(self.ASCII_CHARS, None) 67 | color_coeff = int(color_coeff) 68 | for char in palette: 69 | char_palette = {} 70 | for color in color_palette: 71 | color_key = tuple(color // color_coeff) 72 | char_palette[color_key] = self.font.render(char, False, tuple(color)) 73 | palette[char] = char_palette 74 | return palette, color_coeff 75 | 76 | def get_image(self): 77 | # self.cv2_image = cv2.imread(self.path) 78 | ret, self.cv2_image = self.capture.read() 79 | if not ret: 80 | exit() 81 | transposed_image = cv2.transpose(self.cv2_image) 82 | image = cv2.cvtColor(transposed_image, cv2.COLOR_BGR2RGB) 83 | gray_image = cv2.cvtColor(transposed_image, cv2.COLOR_BGR2GRAY) 84 | return image, gray_image 85 | 86 | def draw_cv2_image(self): 87 | resized_cv2_image = cv2.resize(self.cv2_image, (640, 360), interpolation=cv2.INTER_AREA) 88 | cv2.imshow('img', resized_cv2_image) 89 | 90 | def draw(self): 91 | self.surface.fill('black') 92 | self.draw_converted_image() 93 | # self.draw_cv2_image() 94 | 95 | def save_image(self): 96 | pygame_image = pg.surfarray.array3d(self.surface) 97 | cv2_img = cv2.transpose(pygame_image) 98 | cv2_img = cv2.cvtColor(cv2_img, cv2.COLOR_RGB2BGR) 99 | cv2.imwrite('output/ascii_col_image.jpg', cv2_img) 100 | 101 | def run(self): 102 | while True: 103 | for i in pg.event.get(): 104 | if i.type == pg.QUIT: 105 | exit() 106 | elif i.type == pg.KEYDOWN: 107 | if i.key == pg.K_s: 108 | self.save_image() 109 | if i.key == pg.K_r: 110 | self.record = not self.record 111 | self.record_frame() 112 | self.draw() 113 | pg.display.set_caption(str(self.clock.get_fps())) 114 | pg.display.flip() 115 | self.clock.tick() 116 | 117 | 118 | if __name__ == '__main__': 119 | app = ArtConverter() 120 | app.run() --------------------------------------------------------------------------------