タイムラグを与えて相関関係をとりたい。

Question

動画内のある点に対する他の点の強度の相関関係をマッピングするプログラムを作成しました。強度は（（R+G+B）/３）としています。
コードは以下になります。

”場所を決める”と書かれている所に座標を指定します。この指定した点に対する他の全ての相関関係をマッピングしたプログラムとなっております。

ここで質問なのですが、指定した点に対するタイムラグを与えた他の箇所の相関関係を算出するためにはどのような処理を施せばよいのでしょうか？

どうしてもタイムラグを与えることができません

どなたかご教授よろしくお願い致します。

# -*- coding: utf-8 -*-
import time
#from urllib.request import urlretrieve

import cv2
from numpy import *
import matplotlib.pyplot as plt

start = time.time()

# サンプル用動画をダウンロードする。
video = "clean50_edit_denoise.mkv"
#video_url = "https://github.com/opencv/opencv/raw/master/samples/data/vtest.avi"
#urlretrieve(video_path)

# 場所を決める
#点滅の場所
xpx=int(raw_input("点滅1 X:"))
ypx=int(raw_input("点滅1 Y:"))

lag = int(raw_input("lag (frm):"))


# 動画の全フレームを読み込む。
cap = cv2.VideoCapture(video)
count = cap.get(cv2.CAP_PROP_FRAME_COUNT) # number of frames
# 1フレームずつ取得する。
frames = []
while True:
    ret, frame = cap.read()
    if not ret:
        break
    frames.append(frame)
#print(imgs)
#a=array(imgs)
#print(a.shape)
frames = array(frames)

frames=mean(frames, axis=-1)

# ピクセルごとに相関係数を計算する。
x0, y0 = xpx, ypx  # 対称とする点

h, w = frames.shape[1:3]  # 画像の幅、高さ
vec1 = frames[:, y0, x0]
#print(vec1)
corr_img = empty((h, w))

for y in range(h):
    for x in range(w):
        vec2 = frames[:, y, x]
        # 相関行列を計算する。
        C = corrcoef(vec1, vec2)
        corr_img[y, x] = C[0, 1]  # x, y の相関係数

#print(f"Done! {time.time() - start}s")  # 66.96258449554443s

#tPoints=arange(len(intensity1))*(1/fps)

fig, ax = plt.subplots()
ax.invert_yaxis()
#im=ax.imshow(std,cmap="gray")
ax.set_title("Correlation map ")
#plt.pcolor(std1,cmap="jet")
#plt.imshow(std, cmap="jet",vmin=0,vmax=32)
plt.imshow(corr_img, cmap="jet",vmin=-1,vmax=1)
plt.colorbar()
plt.show()

Answer 1

vec1 = [:, y0, x0]
vec2 = [:, y, x]

はそれぞれ全行取ってきているのでこれを単にラグの分だけずらせばよいのでは？

lag = 10
vec1 = [0:len(vec1)-lag, y0, x0]
vec2 = [lag:len(vec2), y, x]

ラグの取り方がvec1, vec2で逆なら逆に

Comments

[退会済みユーザー]

解答ありがとうございます。
実際にその方法を試して行ってみましたが、
File "correlation_map_only_lag.py", line 45
vec1 = [0:len(vec1)-lag, y0, x0]
^
SyntaxError: invalid syntax
------------------

とスライス部分にsyntaxエラーが生じてしまい計算の実行ができませんでした。