Tracker Optical Flow Lucas-Kanade Pruebas.py

import numpy as np
import cv2 as cv
import Config
import Utiles


# cv.namedWindow("frame", cv.WINDOW_NORMAL)
# cv.setWindowProperty("frame", cv.WND_PROP_FULLSCREEN, cv.WINDOW_FULLSCREEN)

cap = cv.VideoCapture("Entrada/vtest.avi")
# cap = cv.VideoCapture(0)
# params for ShiTomasi corner detection
feature_params = dict(maxCorners=100,
					qualityLevel=0.3,
					minDistance=7,
					blockSize=7)
# Parameters for lucas kanade optical flow
lk_params = dict(winSize=(15, 15),
				maxLevel=2,
				criteria=(cv.TERM_CRITERIA_EPS |
						cv.TERM_CRITERIA_COUNT, 10, 0.03))
# Create some random colors
color = np.random.randint(0, 255, (100, 3))
# Take first frame and find corners in it
ret, old_frame = cap.read()
old_gray = cv.cvtColor(old_frame, cv.COLOR_BGR2GRAY)
p0 = cv.goodFeaturesToTrack(old_gray, mask=None, **feature_params)
print(p0)

p0 = np.array([[[270, 250]]], dtype=np.float32)
print(p0)
# Create a mask image for drawing purposes
mask = np.zeros_like(old_frame)

titulo = "Tracker Lucas-Kanade"
# Configuracion de video
Config.Fullscreen(titulo)
# Declara variables para llenar mas adelante
out = None
dims = (0, 0)
# Para capturar la salida
if Config.VidProp.guardar:
	from Config import VidProp
	out = cv.VideoWriter(f"Salida/{titulo}.avi", VidProp.fourcc,
			VidProp.fps, VidProp.resolu)
dims = Utiles.dimensiones_video(cap)


while cap.isOpened():
	ret, frame = cap.read()
	if not ret:
		break
	frame_gray = cv.cvtColor(frame, cv.COLOR_BGR2GRAY)
	# calculate optical flow
	p1, st, err = cv.calcOpticalFlowPyrLK(old_gray, frame_gray, p0, None,
										**lk_params)
	# Select good points
	good_new = p1[st == 1]
	good_old = p0[st == 1]
	# draw the tracks
	for i, (new, old) in enumerate(zip(good_new, good_old)):
		a, b = new.ravel()
		c, d = old.ravel()
		mask = cv.line(mask, (a, b), (c, d), color[i].tolist(), 2)
		frame = cv.circle(frame, (a, b), 5, color[i].tolist(), -1)
	img = cv.add(frame, mask)
	# cv.imshow("mascara", mask)
	cv.imshow(titulo, img)
	if Config.VidProp.guardar:
		Utiles.guardar(out, img)

	k = cv.waitKey(30) & 0xff
	if k == 27:
		break
	# Now update the previous frame and previous points
	old_gray = frame_gray.copy()
	p0 = good_new.reshape(-1, 1, 2)