主頁 > 知識庫 > opencv 分類白天與夜景視頻的方法

opencv 分類白天與夜景視頻的方法

熱門標(biāo)簽:北瀚ai電銷機(jī)器人官網(wǎng)手機(jī)版 地圖標(biāo)注面積 小蘇云呼電話機(jī)器人 北京電銷外呼系統(tǒng)加盟 所得系統(tǒng)電梯怎樣主板設(shè)置外呼 佛山400電話辦理 儋州電話機(jī)器人 朝陽手機(jī)外呼系統(tǒng) 市場上的電銷機(jī)器人

簡訴

最近有個數(shù)據(jù)需要分類處理,是一批含有白天跟夜晚的視頻數(shù)據(jù),需要進(jìn)行區(qū)分開來,單個視頻嚴(yán)格是只有一個場景的,比如說白天整個視頻就一定是白天,因為數(shù)據(jù)量有些大,幾千個視頻,所以就使用代碼簡單區(qū)分下,最后運行結(jié)果還可以,準(zhǔn)確率百分之80十多,當(dāng)然本批數(shù)據(jù)不用太嚴(yán)格,所以代碼區(qū)分完全夠了。

邏輯

  •  opencv讀取視頻
  • 視頻幀圖片轉(zhuǎn)為灰度值圖片
  • 檢測偏暗元素所占整張圖片的比例,大于一定閾值就認(rèn)為該視頻為黑夜。
  • 選取一部分視頻進(jìn)行判斷,并不是整個視頻跑完。
  • 當(dāng)這部分視頻幀為黑夜占比選取全部視頻幀的50%時認(rèn)為該視頻為黑夜環(huán)境,移動該視頻文件到另外一個文件夾。

結(jié)果

最初先測試9個視頻,100%分類正確。



在進(jìn)行多次閾值預(yù)設(shè)后,選取一個比較合適的閾值進(jìn)行處理,準(zhǔn)確率大概86%左右。

源碼

import cv2
import numpy as np
import os,time
import shutil
def GetImgNameByEveryDir(file_dir,videoProperty):  
    FileNameWithPath = [] 
    FileName         = []
    FileDir          = []
    for root, dirs, files in os.walk(file_dir):  
        for file in files:  
            if os.path.splitext(file)[1] in videoProperty:  
                FileNameWithPath.append(os.path.join(root, file))  # 保存圖片路徑
                FileName.append(file)                              # 保存圖片名稱
                FileDir.append(root[len(file_dir):])               # 保存圖片所在文件夾
    return FileName,FileNameWithPath,FileDir
 
def img_to_GRAY(img,pic_path):
    #把圖片轉(zhuǎn)換為灰度圖
    gray_img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
    #獲取灰度圖矩陣的行數(shù)和列數(shù)
    r,c = gray_img.shape[:2]
    piexs_sum=r*c #整個圖的像素個數(shù)
    #遍歷灰度圖的所有像素
    #灰度值小于60被認(rèn)為是黑
    dark_points = (gray_img  60)
    target_array = gray_img[dark_points]
    dark_sum = target_array.size #偏暗的像素
    dark_prop=dark_sum/(piexs_sum) #偏暗像素所占比例
    if dark_prop >=0.60: #若偏暗像素所占比例超過0.6,認(rèn)為為整體環(huán)境黑暗的圖片
        return 1
    else:
        return 0

if __name__ =='__main__':
    path="C:\\Users\\Administrator\\Desktop\\cut_video"
    new_path=path+"\\DarkNight"
    if not os.path.exists(new_path):
        os.mkdir(new_path)
    FileName,FileNameWithPath,FileDir=GetImgNameByEveryDir(path,'.mp4')
    for i in range(len(FileNameWithPath)):
        video_capture = cv2.VideoCapture(FileNameWithPath[i])
        video_size = (int(video_capture.get(cv2.CAP_PROP_FRAME_WIDTH)),int(video_capture.get(cv2.CAP_PROP_FRAME_HEIGHT)))
        total_frames = int(video_capture.get(cv2.CAP_PROP_FRAME_COUNT))
        video_fps = int(video_capture.get(5))
        start_fps=2*video_fps #從2秒開始篩選
        end_fps=6*video_fps #6秒結(jié)束
        avg_fps=end_fps-start_fps #總共fps
        video_capture.set(cv2.CAP_PROP_POS_FRAMES, start_fps) #設(shè)置視頻起點
        new_paths=new_path+"\\"+FileName[i]
        j=0
        count=0
        while True:
            success,frame = video_capture.read()
            if success:
                j += 1
                if(j>=start_fps and j = end_fps):
                    flag=img_to_GRAY(frame,FileNameWithPath[i])
                    if flag==1:
                        count+=1
                elif(j>end_fps):
                    break
            else:
                break
        print('%s,%s'%(count,avg_fps))
        if count>int(avg_fps*0.48): #大于fps50%為黑夜
            print("%s,該視頻為黑夜"%FileNameWithPath[i])
            video_capture.release() #釋放讀取的視頻,不占用視頻文件
            time.sleep(0.2)
            shutil.move(FileNameWithPath[i],new_paths)
        else:
            print("%s,該視頻為白天"%FileNameWithPath[i])

到此這篇關(guān)于opencv 分類白天與夜景視頻的方法的文章就介紹到這了,更多相關(guān)opencv 分類白天與夜景視頻內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家!

您可能感興趣的文章:
  • python使用opencv按一定間隔截取視頻幀
  • opencv 獲取rtsp流媒體視頻的實現(xiàn)方法
  • Python OpenCV獲取視頻的方法
  • 使用Python opencv實現(xiàn)視頻與圖片的相互轉(zhuǎn)換
  • 對Python+opencv將圖片生成視頻的實例詳解
  • python opencv讀mp4視頻的實例
  • Python OpenCV讀取顯示視頻的方法示例
  • Java使用OpenCV3.2實現(xiàn)視頻讀取與播放
  • Opencv實現(xiàn)讀取攝像頭和視頻數(shù)據(jù)
  • opencv實現(xiàn)讀取視頻保存視頻
  • opencv3/C++實現(xiàn)視頻讀取、視頻寫入

標(biāo)簽:云南 江蘇 酒泉 定西 金融催收 商丘 寧夏 龍巖

巨人網(wǎng)絡(luò)通訊聲明:本文標(biāo)題《opencv 分類白天與夜景視頻的方法》,本文關(guān)鍵詞  opencv,分類,白天,與,夜景,;如發(fā)現(xiàn)本文內(nèi)容存在版權(quán)問題,煩請?zhí)峁┫嚓P(guān)信息告之我們,我們將及時溝通與處理。本站內(nèi)容系統(tǒng)采集于網(wǎng)絡(luò),涉及言論、版權(quán)與本站無關(guān)。
  • 相關(guān)文章
  • 下面列出與本文章《opencv 分類白天與夜景視頻的方法》相關(guān)的同類信息!
  • 本頁收集關(guān)于opencv 分類白天與夜景視頻的方法的相關(guān)信息資訊供網(wǎng)民參考!
  • 推薦文章