Pythonを使って動画をアスキーアート化してターミナル上で再生したい。

Question

前提・実現したいこと

まずPythonを使用して動画をフレームごとに切り分けて保存します。
その後アスキーアート化し、ターミナル上で上書きし続けながら出力するというものです。
###発生している問題
フレーム全てをアスキーアートへと変えることはできましたが、ターミナル上で上書きしながら出力し続けることができません。
そのためアスキーアートがターミナル上で流れるように出力(1枚1枚新しく表示)されてしまい、見づらいものになってしまいました。

特にこの二行が思った通りに動きません。
・sys.stdout.write("\r" + ascii_image)
・sys.stdout.flush()

###該当のソースコード

#指定したファイルから画像を読み込み、アスキーアートに変換し出力するプログラム
import PIL.Image
import cv2
import sys
import time

ASCII_CHARS = ["@","#","S","%","?","*","+",";",":",",","."]

#リサイズ
def resize_image(image,new_width=100):
    width,height = image.size
    ratio = height / width 
    new_height = int(new_width * ratio)
    resized_image = image.resize((new_width,new_height))
    return(resized_image)

#グレースケール化
def convert_grayscale(image):
    grayscale_image = image.convert("L")
    return(grayscale_image)

#変換
def pixels_to_ascii(image):
    pixels = image.getdata()
    characters = ''.join([ASCII_CHARS[pixel//25] for pixel in pixels]) 
    return(characters)
    

def main(new_width=100):

    for index in range(6565):
        path = "frame/frame_" + str(index).zfill(4) +".jpg"
        
        image = PIL.Image.open(path)
        

        new_image_data = pixels_to_ascii(convert_grayscale(resize_image(image)))

        pixel_count = len(new_image_data)
        ascii_image = "\n".join([new_image_data[i:(i+new_width)] for i in range(0,pixel_count,new_width)]) #ascii化


        sys.stdout.write("\r" + ascii_image)
        sys.stdout.flush()

        time.sleep(0.033)


main()

#こちらは動画からフレームごことに切り出すプログラム
import cv2
import os

def save_all_frames(video_path="動画の名前と拡張子", dir_path="frame", basename="frame", ext='jpg'):
    cap = cv2.VideoCapture(video_path)

    if not cap.isOpened():
        return

    os.makedirs(dir_path, exist_ok=True)
    base_path = os.path.join(dir_path, basename)

    digit = len(str(int(cap.get(cv2.CAP_PROP_FRAME_COUNT))))#総フレーム数の桁を取得

    n = 0

    while True:
        ret, frame = cap.read()
        if ret:
            cv2.imwrite('{}_{}.{}'.format(base_path, str(n).zfill(digit), ext), frame)#画像で出力
            n += 1
        else:
            return

save_all_frames()

####補足
・アスキーアート化させたい画像はカレントディレクトリの一つ下のファイルに保存しています
/frame/frame_0000.jpg (0000から6565までの6566枚)

・1フレームあたりおよそ0.033秒ほどの表示時間で約30fpsほどです。

・参考にした動画 : リンク内容

至らない点も多々あるかと思いますがよろしくお願いします。

Comments

[y_waiwai]

しつもんはなんでしょうか

Answer 1

アスキーアートをコンソール上でアニメーションするコード例をいくつか見てみましたが、どれも各画像の描画の間にコンソール画面を白紙にする処理を入れています。いちど試してみてはいかがでしょうか。

• https://stackoverflow.com/questions/2726343/how-to-create-ascii-animation-in-a-console-application-using-python-3-x
• https://stackoverflow.com/questions/60783120/ascii-animation-on-python
• https://medium.com/sweetmeat/python-curses-based-ascii-art-fire-animation-259e9e007767

それと画像の描画ごとに画像ファイルから読み込んでリサイズして白黒化してアスキーアート化する処理を行っていますが、負荷がどの程度なのか気になります。処理にかかる時間によっては思ったより画像の切り替えにモタったりすることもあるのではないかと思うのですが。
リアルタイム動画ではないので、先にテキストファイルに書き出しておいてから、描画の時はそこから読み込んでもいいのではないかと思います。
もっともこれは試してみたわけではないので、見当はずれかも知れません。

Comments

[fiaro-san]

一度コンソールを白紙にする処理を追加したところ思い通りに動くようになりました。ありがとうございます。