功能性模組：（5）圖片生成視訊：ffmpeg版和OpenCV版

一、模組介紹

這個模組其實是非常常用、基本的模組了，主要就是針對一些需要依靠視訊才能評判效果的一些應用，當然小夥伴們按自己的需求來吧。

這個圖片呢其實是LZ事先處理好的，這個裡有一個小技巧可以分享給小夥伴們，在影像命名的時候儘量以00001.jpg，00002.jpg這種方式來命名，這樣對影像進行排序的時候也會更加方便，不至於出現1.jpg，10.jpg，2.jpg…這種情況，會非常麻煩。

二、程式碼實現

2.1 OpenCV版本

OpenCV的安裝LZ已經前前後後安裝好幾十次了，包括c++版本，cuda版本，python版本都安裝了好多次了，這個就不再贅述了。

OpenCV有一個比較好的是即使你的影像命名是不連續的，也都可以寫入到視訊中去，LZ覺得還是比較方便的。

 # generate video using OpenCV
def GenVideoFromImg(frame_root, output_video_path):
    '''
    :param frame_root: 存放圖片打路徑，這裡可以根據需要自己修改
    :param output_video_path: 輸出視訊的名稱，
    :return:
    '''
    img_lists = glob.glob(frame_root + "/*.jpg")
    print("Image path is: ", frame_root)
    print("The length of img_lists: ", len(img_lists))
    img_lists.sort()
    img_test = cv2.imread(img_lists[0])
    h, w, c = img_test.shape
    fourcc = cv2.VideoWriter_fourcc('X', 'V', 'I', 'D')
    # 為了仔細看效果，幀率調的很小
    video_writer = cv2.VideoWriter(output_video_path, fourcc, 10, (w, h))
    print("Begin to generate video!")
    for img_list in tqdm(img_lists):
        img = cv2.imread(img_list)
        video_writer.write(img)
    video_writer.release()

2.2 ffmpeg版

ffmpeg版本其實LZ並沒有很會使用，但是為什麼要使用呢？是因為目前LZ公司的資料（包括測試資料）是沒有辦法下載到本地的，並且訓練的模型也沒法下載到本地，這樣就需要在雲平臺上進行測試，但是雲平臺的播放只支援h264的播放，所以LZ只能找對應的方法了。

2.2.1 安裝對應包

ffmpeg command not found

出現這個問題其實就是沒有對應依賴包，安裝一下就行了，因為就是為了生成對應的視訊，所以對於版本沒有什麼很高的要求

sudo apt install ffmpeg
#或者
sudo apt-get install ffmpeg

2.2.2 程式碼實現

首先會看到一個ReName的函式，這個主要是為了整理一下圖片的名稱，以為ffmpeg對應圖片幀是要對應圖片名字是連續的，這個問題LZ找了好久，因為是跳幀進行測試的，所以圖片名稱是00001.jpg，00003.jpg。。。LZ生成了半天視訊，發現始終只有一幀影像，倒騰了好久，才發現這個問題，哭暈。。還要自己給個許可權，視訊才能播放。

def ReName(img_path):
    """

    :param img_path: 視訊序列的路徑
    :return:
    """
    img_lists = os.listdir(img_path)
    img_lists.sort()
    count = 0
    for i in range(0, len(img_lists)):
        idx = int(img_lists[i].split('.')[0])
        os.rename(os.path.join(img_path, img_lists[i]),
                  os.path.join(img_path, str(("%05d" % count)) + '.' + 'jpg'))
        count = count + 1

    # print(img_lists)


def ConvertVideoH264(video_root, output_dir, seq):
    '''

    :param video_root: 視訊儲存的總的路徑
    :param output_dir: 儲存圖片的總的路徑
    :param seq: 視訊名稱，不含字尾名
    :return:
    '''
    ReName("{}/frame/".format(output_dir))
    output_video_path = osp.join(output_dir, '{}.mp4'.format(seq))
    output_video_path_h264 = osp.join(video_root, '{}-h264.mp4'.format(seq))
    cmd_str = 'ffmpeg -r 10 -f image2 -i {}/frame/%05d.jpg -c:v copy {}'.format(output_dir, output_video_path)
    os.system(cmd_str)
    cmd_convert = 'ffmpeg -i {} -vcodec h264  {}'.format(output_video_path, output_video_path_h264)
    os.system(cmd_convert)
    os.chmod(output_dir, 0o755)

2020年12月29日，最近全國各個地方都開始零零星星的出現新冠疫情，真的希望這場疫情能快點過去，脫下口罩，看遍祖國大好河山，加油！