Python opencv imshow()の代わりにpyplotのimshow()を使って動画をフレームごとに再生したい。

Question

pyplotのメソッドを使ってフレームごとに動画を再生したい。

cv2のメソッドを使いながら色々試しているとcv2.imshow()が使えないという難題に直面しました。
色々試しましたが解決できそうにないので、pyplotのimshow()メソッドで代用することにしました。しかし、画像一枚だけを表示するのなら簡単にできますが、動画のフレームごとに止まることなく再生させるすべがわかりません。
もし詳しい方がいれば教えていただけると幸いです。
フレームごとに処理を加えたいのでそれが可能なコード方法を教えて頂きたいです。

試しに書いたコード（qを押さないと次のフレームに進んでくれない）

import cv2 as cv
from matplotlib import pyplot as plt

filepath = "Videos/CatVideo.mp4"

# 動画の読み込み
cap = cv.VideoCapture(filepath)

# 動画終了まで繰り返し
while(cap.isOpened()):
    # フレームを取得
    ret, frame = cap.read()

    # フレームを表示
    plt.imshow(frame)
    plt.show()
    #! cv.show()は使えないのでpyplotで代用
    # cv.startWindowThread()
    # cv.namedWindow("preview")
    # cv.imshow("preview", frame)

    # qキーを押さないと次のフレームに進んでくれない。
    if cv.waitKey(1) & 0xFF == ord('q'):
        break

cap.release()
cv.destroyAllWindows

よろしくお願いします。

Comments

[jbpb0]

https://github.com/tody411/PyIntroduction/blob/master/opencv/pycv_tutorial/video_capture.md
の後半の「pltCaputreVideo()」みたいなのですか？

もしそうなら、
animation.FuncAnimation
でググれば、使用例いろいろ見つかります

[RoundCat]

ありがとうございます！とてもいいヒント頂きました！いろいろと調べて見ますね！

[jbpb0]

上記Webサイトのコードにもありますように、Kenta_pyさんが回答に書かれた二つ目のコードの
ani.save(...
の行を
plt.show()
に変えたら、ファイル保存ではなく画面表示になります

また、上記行の後に、
cap.release()
を入れた方がいいかも

どちらも既にご存知だとは思いますが、念の為に追記しておきます

Answer 1

import cv2
import sys
file_path = "cup.mp4"
# 動画の読み込み
delay = 1
window_name = 'frame'
cap = cv2.VideoCapture(file_path)
cap.isOpened()
if not cap.isOpened():
    sys.exit()

while True:
    ret, frame = cap.read()
    if ret:
        cv2.imshow(window_name, frame)
        if cv2.waitKey(delay) & 0xFF == ord('q'):
            break
    else:
        cap.set(cv2.CAP_PROP_POS_FRAMES, 0)

cv2.destroyWindow(window_name)

こんにちは、ここのコードを参考にしました。うまく動画を表示することができたので、こちらでうまくいくか試してみてください。

import matplotlib.animation as anm
import matplotlib.pyplot as plt
import numpy as np
import cv2
import matplotlib.pyplot as plt

file_path = "cup.mp4"
cap =cv2.VideoCapture(file_path)
def update(i, fig_title, A):
    if i != 0:
        plt.cla()                      # 現在描写されているグラフを消去

    ret,frame = cap.read()
    frame=cv2.cvtColor(frame,cv2.COLOR_BGR2RGB)
    plt.clf()
    plt.imshow(frame)

fig = plt.figure(figsize=(4,3), dpi=100)
ani = anm.FuncAnimation(fig, update, fargs = ('Initial Animation! ', 2.0), \
    interval = 100, frames = 132)
ani.save("Sample.gif", writer = 'imagemagick')

2つ目のものは、このページをもとに作成しましたが、うまくいくか自信がありません。

Comments

[RoundCat]

回答して頂けたのは非常に有難いのですが、cv2.imshow()の関数は僕のパソコン正常に動いてくれないので使えないんです。。

[Kenta_py]

なるほど、そうなんですね、失礼いたしました。それでpltによる方法が必要なんですね。ちなみに、私のコードでも同様にエラーが出ましたか？

[RoundCat]

追記ありがとうございます。いえ、エラーじゃなくてプロンプトが帰って来なくなるんですよね。cv2のimshow()は。
新しく書いて頂いたコード試してみて上手くいったらまたすぐに報告します！

[RoundCat]

Kentaさんから提案いただいたコードを試しました。いくつか修正を加えまして、
１：まずplt.imshow(frame)の前の行にplt.axis("off")を加えてグラフっぽさをなくしたのと、
2 : ani.save("Sample.gif", writer='pillow') とwriterの部分を変更しました。

結果として、gif動画を問題なく書き出すことができました。もう少しコード改善の余地はありそうですが、これでなんとかcv2.imshow()の代わりにはなりそうです。ありがとうございました！！

[Kenta_py]

ありがとうございます。よかったです！
ちなみに
「試しに書いたコード（qを押さないと次のフレームに進んでくれない）」のところの、
# qキーを押さないと次のフレームに進んでくれない。
if cv.waitKey(1) & 0xFF == ord('q'):
break
の２行を消去したうえでplt.imshowで表示させるとどうなるんでしょうか。

[RoundCat]

あけましておめでとうございます！
その２行を消しても変わりありません！qを押すか、ウィンドウを閉じるボタンを押すかしないと次のフレームには進んでくれないです。
plt.show()ではどうやらqを押すか、ウィンドウを閉じるかしないと処理が終わらないようです。。