michaelfsp
/
bigmaze


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
30

	
31

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
41

	
42

	
43

	
44

	
45

	
            import ffmpeg
import pickle
import os

whole_vid_dir = '/media/hdd_data1/michael/deeplabcut'
videos = {'15122014': 'crop_30fps2014-12-15T15_42_21.avi',
          '16122014': 'crop_30fps2014-12-16T15_35_28.avi',
          '17122014': 'crop_30fps2014-12-17T17_23_38.avi',
          '18122014': 'crop_30fps2014-12-18T18_09_05.avi',
          '19122014': 'crop_30fps2014-12-19_concatenated.avi',
          '20122014': 'crop_30fps2014-12-20T17_53_17.avi',
          '21122014': 'crop_30fps2014-12-21T15_10_39.avi',
          '22122014': 'crop_30fps2014-12-22T16_51_00.avi',
          '23122014': 'crop_30fps2014-12-23T14_08_57.avi'}

video_out_dir = '/media/hdd_data1/michael/deeplabcut/subvideos'

mice = ['4I', '4II', '4III', '4IV', '5I', '5II', '5III', '5IV']
days = list(videos.keys())

# file containing start and end times for each mice in the session video in ms
metadata_file_path = '../cache/time_shifts.pickle'


with open(metadata_file_path, 'rb') as f:
    times = pickle.load(f)

# TODO: this should be substituted by a for loop for all days
for day in days:
    video_file_path = os.path.join(whole_vid_dir, videos[day])
    stream = ffmpeg.input(video_file_path)

    for mouse in mice:
        t_start, t_stop = times[day][mouse]  # in ms
        t_start /= 1000  # convert to s
        t_stop /= 1000  # convert to s
        stream = ffmpeg.input(video_file_path, ss=t_start, to=t_stop)
        # stream = ffmpeg.trim(stream, start=t_start, end=t_stop)
        video_out_dir_day = os.path.join(video_out_dir, day)
        if not os.path.exists(video_out_dir_day):
            os.makedirs(video_out_dir_day)
        video_out_path = os.path.join(video_out_dir_day,
                                      'crop_30fps_%s_%s.avi' % (day, mouse))
        stream = ffmpeg.output(stream, video_out_path, codec='copy')
        ffmpeg.run(stream)