前提・実現したいこと
pythonでpcのカメラで動画を撮影しながら、勾配計算をした画像(laplacian)と背景差分計算をした画像(fgmask)に縦横16pxごとに分けるプログラムを作っています。
背景の画像は、黒でマスクしてエッジを赤色にしています。
下記のエラーメッセージが出てきますが、どうして30が範囲外になってしまうのかがわかりません。
そこを解説していただきたいです。
発生している問題・エラーメッセージ
if fgmask_sum[x][y] >= fgmask_mean: IndexError: index 30 is out of bounds for axis 0 with size 30
該当のソースコード
python
import numpy as np import cv2 REGION_WIDTH=16 REGION_HIGH=16 cap = cv2.VideoCapture(0) cap.set(cv2.CAP_PROP_FRAME_WIDTH, 640) cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 480) fgbg = cv2.createBackgroundSubtractorMOG2(history=500, varThreshold=16, detectShadows=True) red_mask = np.full((1, 1, 3), (0,0,255), dtype=np.uint8) white_mask = np.full((REGION_HIGH, REGION_WIDTH, 3), (255,255,255), dtype=np.uint8) while(True): ret, frame = cap.read() rows, cols, channels = frame.shape # maskを定義 laplacian_mask = np.zeros((rows, cols, channels), dtype=np.uint8) fgmask_mask = np.zeros((rows, cols, channels), dtype=np.uint8) #エッジを赤くする frame_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) laplacian = cv2.Laplacian(frame_gray, cv2.CV_64F) for y in range(rows): for x in range(cols): if laplacian[y,x] > 10: laplacian_mask[y,x] = np.copy(red_mask) #マスクしない(白色でマスク)部分を組み込む fgmask = fgbg.apply(frame) fgmask_sum = np.zeros((30,40))#16*16 ごとに領域分けした時の領域の数 for i in range(int(rows/REGION_HIGH)):#30 for j in range(int(cols/REGION_WIDTH)):#40 for y in range(REGION_HIGH):#16 for x in range(REGION_WIDTH):#16 if (fgmask[i*16+y][j*16+x]==255):#1画素ずつ白かどうかの判定 fgmask_sum[i,j] += 1 fgmask_mean = np.mean(fgmask_sum)#全領域の白の要素の平均 for y in range(int(rows/REGION_HIGH)):#30 for x in range(int(cols/REGION_WIDTH)):#40 if fgmask_sum[x][y] >= fgmask_mean:#エラーが発生している行 fgmask_mask[16*y:16*y+16,16*x:16*x+16] = np.copy(white_mask) #表示 mask_or = cv2.bitwise_or(laplacian_mask, fgmask_mask) masked_frame = cv2.bitwise_and(frame, mask_or) cv2.imshow('frame', masked_frame) k = cv2.waitKey(30) & 0xff if k == 27: break # 'ESC' key is pressed cap.release() cv2.destroyAllWindows()
まだ回答がついていません
会員登録して回答してみよう