Coverage for src/bob/bio/video/database/database.py: 100%

1from bob.bio.base.database.file import BioFile 

2 

3from ..utils import VideoAsArray 

4 

5 

6class VideoBioFile(BioFile): 

7 def __init__( 

8 self, 

9 client_id, 

10 path, 

11 file_id, 

12 original_directory=None, 

13 original_extension=".avi", 

14 annotation_directory=None, 

15 annotation_extension=None, 

16 annotation_type=None, 

17 selection_style=None, 

18 max_number_of_frames=None, 

19 step_size=None, 

20 **kwargs, 

21 ): 

22 """ 

23 Initializes this File object with an File equivalent for 

24 VoxForge database. 

25 """ 

26 super().__init__( 

27 client_id=client_id, 

28 path=path, 

29 file_id=file_id, 

30 original_directory=original_directory, 

31 original_extension=original_extension, 

32 annotation_directory=annotation_directory, 

33 annotation_extension=annotation_extension, 

34 annotation_type=annotation_type, 

35 **kwargs, 

36 ) 

37 self.selection_style = selection_style or "all" 

38 self.max_number_of_frames = max_number_of_frames 

39 self.step_size = step_size 

40 

41 def load(self): 

42 path = self.make_path(self.original_directory, self.original_extension) 

43 return VideoAsArray( 

44 path, 

45 selection_style=self.selection_style, 

46 max_number_of_frames=self.max_number_of_frames, 

47 step_size=self.step_size, 

48 )