Opencvで動画を保存したい

Question

前提・実現したいこと

連投失礼します。
PythonでOpencvライブラリを用いて、入力動画にガンマ補正をかけて明るく見やすくなるように動画として出力したいです。
ファイル形式に対してのこだわりは特にありません。
今回は一応avi形式で出力しようとしています。
出力動画の高さ、幅、fpsは入力動画のそれらに合わせます。

発生している問題・エラーメッセージ

エラーは出力されずに処理は終了します。
ただ、出力されたファイルが正しくフレームが書き込まれておらず、ファイルを再生することができません。
データサイズも入力ファイルが50MBほどあるにも関わらず、6kBとなっています。

該当のソースコード

ソースコード
#ライブラリ等のインストールは済ませてあります。
#ガンマ補正後に動画を保存する関数
def save_file(video_data,gamma):
    cap = cv2.VideoCapture(video_data)
    #動画のコーディックを指定するコード
    fourcc = cv2.VideoWriter_fourcc(*'XVID')
    #動画像の書き込み仕様設定（動画の再生速度と解像度を指定）
    fps,height,width = video_info(video_data)
    out = cv2.VideoWriter('gamma{}_output.avi'.format(gamma),fourcc, fps, (int(width),int(height)))
    #ガンマテーブル作成
    gamma_cvt = np.zeros((256,1),dtype = 'uint8')
    for i in range(256):
        gamma_cvt[i][0] = 255 * (float(i)/255) ** (1.0/gamma)
    while(True):
        ret, img = cap.read()
        if ret==True:
            #グレースケール変換
            frame = cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)
            frame = cv2.LUT(frame,gamma_cvt)
            out.write(frame)
            cv2.imshow('frame',frame)
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
        else:
            break
    # Release everything if job is finished
    cap.release()
    out.release()
    cv2.destroyAllWindows()

#ビデオの基本データ出力
#ビデオデータのfps,高さ、幅を出力してくれる。さらにグローバル変数でさるfps,height,widthにも同時に代入される
def video_info(video_data):
    movie = cv2.VideoCapture(video_data) 
    fps    = movie.get(cv2.CAP_PROP_FPS)
    height = movie.get(cv2.CAP_PROP_FRAME_HEIGHT)
    width  = movie.get(cv2.CAP_PROP_FRAME_WIDTH)
    print("fpsは{}".format(fps))
    print("heightは{}".format(height))
    print("widthは{}".format(width))
    return fps,height,width

VIDEO_DATA = "test_movie.avi"
save_file(VIDEO_DATA,2.0)

試したこと

コーディックに問題があるのかと思い、DIVX,XVID,WMV1など試しましたが、どのコーディック指定をしても生成できる形でファイルは出力されませんでした。

補足情報（FW/ツールのバージョンなど）

実行環境
opencv 3.3.1
jupyter notebook
windows 10

ヒント等でもいいので、何か解決方法に心当たり等ありましたら教えてください。

Answer 1

こちら試行錯誤の末、解決しました。
どうやら、書き出しファイルがカラー指定なのにグレースケール変換してしまったことが原因のようでした。
具体的には、以下コードの変更の必要性がありました。

out = cv2.VideoWriter('gamma{}_output.avi'.format(gamma),fourcc, fps, (int(width),int(height)))

cv2.VideoWriterには第５引数も実はあり、TrueだとRGB、Falseだとグレースケールでの保存がされるようです。
なので、グレースケールに変換する場合は、第５引数に明示的にFalseを記述する必要性があるようです、

Answer 2

ret, img = cap.read()のretがTrueではないんじゃないでしょうか。

Comments

[kentashx]

いえ，cv2.imshowの出力を確認することでretがTrueであることは確認できています．