1- from tqdm import tqdm as show_progress
2- import cv2
1+ # from tqdm import tqdm as show_progress
2+ # import cv2
33
4- from ..scene .scene import Scene
5- from ..logger import logger
4+ # from ..scene.scene import Scene
5+ # from ..logger import logger
66
77
8- # TODO, is this depricated?
9- class SceneFromVideo (Scene ):
10- def construct (self , file_name ,
11- freeze_last_frame = True ,
12- time_range = None ):
13- cap = cv2 .VideoCapture (file_name )
14- self .shape = (
15- int (cap .get (cv2 .cv .CV_CAP_PROP_FRAME_HEIGHT )),
16- int (cap .get (cv2 .cv .CV_CAP_PROP_FRAME_WIDTH ))
17- )
18- fps = cap .get (cv2 .cv .CV_CAP_PROP_FPS )
19- self .camera .frame_rate = fps
20- frame_count = int (cap .get (cv2 .cv .CV_CAP_PROP_FRAME_COUNT ))
21- if time_range is None :
22- start_frame = 0
23- end_frame = frame_count
24- else :
25- start_frame , end_frame = [fps * t for t in time_range ]
8+ # # TODO, is this depricated?
9+ # class SceneFromVideo(Scene):
10+ # def construct(self, file_name,
11+ # freeze_last_frame=True,
12+ # time_range=None):
13+ # cap = cv2.VideoCapture(file_name)
14+ # self.shape = (
15+ # int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_HEIGHT)),
16+ # int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_WIDTH))
17+ # )
18+ # fps = cap.get(cv2.cv.CV_CAP_PROP_FPS)
19+ # self.camera.frame_rate = fps
20+ # frame_count = int(cap.get(cv2.cv.CV_CAP_PROP_FRAME_COUNT))
21+ # if time_range is None:
22+ # start_frame = 0
23+ # end_frame = frame_count
24+ # else:
25+ # start_frame, end_frame = [fps * t for t in time_range]
2626
27- frame_count = end_frame - start_frame
28- logger .info ("Reading in " + file_name + "..." )
29- for count in show_progress (list (range (start_frame , end_frame + 1 ))):
30- returned , frame = cap .read ()
31- if not returned :
32- break
33- # b, g, r = cv2.split(frame)
34- # self.frames.append(cv2.merge([r, g, b]))
35- self .frames .append (frame )
36- cap .release ()
27+ # frame_count = end_frame - start_frame
28+ # logger.info("Reading in " + file_name + "...")
29+ # for count in show_progress(list(range(start_frame, end_frame + 1))):
30+ # returned, frame = cap.read()
31+ # if not returned:
32+ # break
33+ # # b, g, r = cv2.split(frame)
34+ # # self.frames.append(cv2.merge([r, g, b]))
35+ # self.frames.append(frame)
36+ # cap.release()
3737
38- if freeze_last_frame and len (self .frames ) > 0 :
39- self .original_background = self .background = self .frames [- 1 ]
38+ # if freeze_last_frame and len(self.frames) > 0:
39+ # self.original_background = self.background = self.frames[-1]
4040
41- def apply_gaussian_blur (self , ksize = (5 , 5 ), sigmaX = 5 ):
42- self .frames = [
43- cv2 .GaussianBlur (frame , ksize , sigmaX )
44- for frame in self .frames
45- ]
41+ # def apply_gaussian_blur(self, ksize=(5, 5), sigmaX=5):
42+ # self.frames = [
43+ # cv2.GaussianBlur(frame, ksize, sigmaX)
44+ # for frame in self.frames
45+ # ]
4646
47- def apply_edge_detection (self , threshold1 = 50 , threshold2 = 100 ):
48- edged_frames = [
49- cv2 .Canny (frame , threshold1 , threshold2 )
50- for frame in self .frames
51- ]
52- for index in range (len (self .frames )):
53- for i in range (3 ):
54- self .frames [index ][:, :, i ] = edged_frames [index ]
47+ # def apply_edge_detection(self, threshold1=50, threshold2=100):
48+ # edged_frames = [
49+ # cv2.Canny(frame, threshold1, threshold2)
50+ # for frame in self.frames
51+ # ]
52+ # for index in range(len(self.frames)):
53+ # for i in range(3):
54+ # self.frames[index][:, :, i] = edged_frames[index]
0 commit comments