tanner/min-dalle-test
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

gray_pixels not rgb_pixels

Browse Source
This commit is contained in:
Brett Kuprel 2022-07-14 11:28:03 -04:00
parent 3698105149
commit cc78846506
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
image_from_text.py
View File

@ -17,9 +17,9 @@ parser.add_argument('--top_k', type=int, default=256)
def ascii_from_image(image: Image.Image, size: int = 128) -> str: def ascii_from_image(image: Image.Image, size: int = 128) -> str:
rgb_pixels = image.resize((size, int(0.55 * size))).convert('L').getdata() gray_pixels = image.resize((size, int(0.55 * size))).convert('L').getdata()
chars = list('.,;/IOX') chars = list('.,;/IOX')
chars = [chars[i * len(chars) // 256] for i in rgb_pixels] chars = [chars[i * len(chars) // 256] for i in gray_pixels]
chars = [chars[i * size: (i + 1) * size] for i in range(size // 2)] chars = [chars[i * size: (i + 1) * size] for i in range(size // 2)]
return '\n'.join(''.join(row) for row in chars) return '\n'.join(''.join(row) for row in chars)