projeto_lab/src/modulos/segmentation.py

31 lines
486 B
Python

# TODO
import numpy as np
import cv2 as cv
from sistema.video import Video
class Segmentation(object):
def __init__(self):
pass
def depth_color_map(self, frame):
return frame
def debug(self):
v = Video()
while True:
frame = v.serialize()
frame = self.depth_color_map(frame)
cv.imshow('Debug', frame)
if cv.waitKey(1) == ord('q'):
break
v.close()