OpenCVで動画をはき出す方法。C++のソースコードのどこを改良すればよいのかわからない。

Question

このサイトにあるvideostab.cppのソースコードは入力動画のブレを補正した動画を出力してくれるものです。コンパイルして、

./a.out 動画名.avi

と実行すると、補正前の動画と補正後の動画が並んで表示されます。補正後の動画のみを保存したいのですが、どのようにソースコードを改良したら良いのでしょうか。あいにく、C++は触ったことがなくて、ちんぷんかんぷんで困っています。

いろいろと調べてみた結果、72行目に

VideoWriter writer("x3.avi", CV_FOURCC_DEFAULT, 25, cv::Size(480, 640), true);

を加えて、

264行目に

writer.write(canvas);

を加えてみたのですが、生成された動画ファイルx3.aviは開けませんでした。どなたかお知恵を貸していただけると幸いです。

よろしくお願いします。

Answer 1

実際に画像の補正を行っているのはSTEP5のループです。
また、『補正後の画像』は、canvasではなくcur2です。

VideoWriter writer("x3.avi", CV_FOURCC_DEFAULT, 25, cv::Size(480, 640), true);

// 動画の再生位置を最初に戻す
cap.set(CV_CAP_PROP_POS_FRAMES, 0);

略

// 最後以外の全てのフレームについて
while(k < max_frames-1) {
    // フレームを取得
    cap >> cur;

    略

    // 歪みを補正したものを、cur2に格納
    Mat cur2;
    warpAffine(cur, cur2, T, cur.size());

    // cur2の形を少し整える
    cur2 = cur2(Range(vert_border, cur2.rows-vert_border), Range(HORIZONTAL_BORDER_CROP, cur2.cols-HORIZONTAL_BORDER_CROP));
    resize(cur2, cur2, cur.size());

    // ビデオに書き出し
    writer << cur2;

    略
}

未検証ですが、こんな感じでできるかと思います。

Comments

[physics303]

ありがとうございます。補正後の画像はcur2、補正前の画像はcur、それを結合させたものが、canvasということですかね。実際、手を加えるのは、

VideoWriter writer("x3.avi", CV_FOURCC_DEFAULT, 25, cv::Size(480, 640), true);

と

writer << cur2;

ですね。早速、ためしてみました。コンパイルはうまくいきますし、プログラムも問題なく動作するのですが、生成された動画ファイルx3.aviが開けないのです。Ubuntu環境ですが、「出力ストリームを取り出せません」とエラーが出ます。VLCメディアプレーヤーでも開くことができませんでした。

原因は何でしょうか。

[LouiS0616]

画像のサイズが指定と合わないとそんな感じのエラーが出るようです。
読みこみ元のaviファイルのサイズは640x480で間違いないのでしょうか？

[physics303]

読み込み元のaviファイルのサイズは640×480で間違いないです。幅が640、高が480ですね。

[physics303]

VideoWriter writer("x3.avi", CV_FOURCC_DEFAULT, 25, cv::Size(480, 640), true);

を

VideoWriter writer("x.avi", VideoWriter::fourcc('D', 'I', 'V', '3'), 15.0, Size(640,480));

で書き換えたところ、うまく行きました。
何が問題だったのかはわかりません。。。