68 lines
2.4 KiB
Python
68 lines
2.4 KiB
Python
import cv2
|
|
from video_loader import VideoLoader, VideoType
|
|
from background_heatmap import BackgroundHeatmap, BackgroundHeatmapFromGroup
|
|
import numpy as np
|
|
|
|
|
|
if __name__ == '__main__':
|
|
|
|
params = cv2.SimpleBlobDetector_Params()
|
|
params.minThreshold = 60
|
|
params.maxThreshold = 255
|
|
# Filter by Area.
|
|
params.filterByArea = True
|
|
params.minArea = 20
|
|
params.maxArea = 25000
|
|
params.filterByCircularity = False
|
|
params.filterByColor = False
|
|
params.filterByConvexity = False
|
|
params.filterByInertia = False
|
|
params.blobColor = 255
|
|
|
|
detector = cv2.SimpleBlobDetector_create(params)
|
|
|
|
projects_path = "C:\\UntrackedGit\\opencv_test\\him_projects"
|
|
loader = VideoLoader(projects_path)
|
|
groups = loader.get_recordings_grouped()
|
|
# out = cv2.VideoWriter('outpy.avi',cv2.VideoWriter_fourc\c('M','J','P','G'), 30, (512, 424), isColor=False)
|
|
for group in groups:
|
|
heatmaps = []
|
|
groupheatmap = BackgroundHeatmapFromGroup(group)
|
|
for recording in group[1]:
|
|
first = True
|
|
frames = VideoLoader.extract_frames(recording, video_type=VideoType.DEPTH)
|
|
for frame in frames:
|
|
groupheatmap.update(frame)
|
|
cv2.imshow("IR", frame)
|
|
# out.write(converted)
|
|
# cv2.imshow("Heatmap", heatmap.heatmap)
|
|
cv2.imshow("Background filter", groupheatmap.lastframe)
|
|
diff = groupheatmap.bgf_diff
|
|
|
|
# Erode dilate
|
|
kernel = np.ones((5,5),np.uint8)
|
|
diff = cv2.morphologyEx(diff, cv2.MORPH_OPEN, kernel)
|
|
|
|
keypoints = detector.detect(diff)
|
|
frame_with_keypoints = cv2.drawKeypoints(diff, keypoints, np.array([]), (0,0,255), cv2.DRAW_MATCHES_FLAGS_DRAW_RICH_KEYPOINTS)
|
|
|
|
cv2.imshow("BGF diff", frame_with_keypoints)
|
|
cv2.moveWindow("Background filter", 600, 100)
|
|
cv2.moveWindow("BGF diff", 1200, 100)
|
|
if cv2.waitKey(1000) & 0xFF == ord('q'):
|
|
break
|
|
heatmaps.append(groupheatmap.heatmap)
|
|
# Reset current heatmap and sum
|
|
groupheatmap.reset()
|
|
cv2.destroyAllWindows()
|
|
|
|
for i, bgh in enumerate(heatmaps):
|
|
imname = "Heatmap " + str(i)
|
|
cv2.imshow(imname, bgh)
|
|
cv2.moveWindow(imname, 500 * i, 0)
|
|
cv2.waitKey()
|
|
cv2.destroyAllWindows()
|
|
|
|
# out.release()
|
|
cv2.destroyAllWindows()
|