自学教程：opencv 分类白天与夜景视频的方法

import cv2import numpy as npimport os,timeimport shutildef 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):    #把图片转换为灰度图    gray_img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)    #获取灰度图矩阵的行数和列数    r,c = gray_img.shape[:2]    piexs_sum=r*c #整个图的像素个数    #遍历灰度图的所有像素    #灰度值小于60被认为是黑    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,认为为整体环境黑暗的图片        return 1    else:        return 0if __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秒结束        avg_fps=end_fps-start_fps #总共fps        video_capture.set(cv2.CAP_PROP_POS_FRAMES, start_fps) #设置视频起点        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])