1
- from tqdm import tqdm as show_progress
2
- import cv2
1
+ # from tqdm import tqdm as show_progress
2
+ # import cv2
3
3
4
- from ..scene .scene import Scene
5
- from ..logger import logger
4
+ # from ..scene.scene import Scene
5
+ # from ..logger import logger
6
6
7
7
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]
26
26
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()
37
37
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]
40
40
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
+ # ]
46
46
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