-
Notifications
You must be signed in to change notification settings - Fork 0
/
save_movie_stills.py
50 lines (32 loc) · 1.61 KB
/
save_movie_stills.py
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
46
47
48
49
50
from play_array_list import *
def save_imgs_landing(movie_dataset):
movie = movie_dataset.movies['20101101_C001H001S0019']
im = movie.frames[150].uimg
plt.imsave('flight', im, cmap=plt.get_cmap('gray'))
legextensionframe = movie.legextensionrange[0] - movie.firstframe_ofinterest
im = movie.frames[legextensionframe].uimg
plt.imsave('leg extension', im, cmap=plt.get_cmap('gray'))
im = movie.frames[movie.landingframe_relative].uimg
plt.imsave('landing', im, cmap=plt.get_cmap('gray'))
im = movie.frames[-1].uimg
plt.imsave('sitting', im, cmap=plt.get_cmap('gray'))
def save_imgs_crash(movie_dataset):
movie = movie_dataset.movies['20101111_C001H001S0032']
im = movie.frames[580].uimg
plt.imsave('crash_flight', im, cmap=plt.get_cmap('gray'))
im = movie.frames[780].uimg
plt.imsave('crash_collision', im, cmap=plt.get_cmap('gray'))
im = movie.frames[900].uimg
plt.imsave('crash_legextension', im, cmap=plt.get_cmap('gray'))
im = movie.frames[-1].uimg
plt.imsave('crash_sitting', im, cmap=plt.get_cmap('gray'))
def save_imgs_wingcrash(movie_dataset):
movie = movie_dataset.movies['20101111_C001H001S0040']
im = movie.frames[-148+646].uimg
plt.imsave('wingcrash_flight', im, cmap=plt.get_cmap('gray'))
im = movie.frames[-89+646].uimg
plt.imsave('wingcrash_collision', im, cmap=plt.get_cmap('gray'))
im = movie.frames[55+646].uimg
plt.imsave('wingcrash_legext', im, cmap=plt.get_cmap('gray'))
im = movie.frames[-1].uimg
plt.imsave('sitting', im, cmap=plt.get_cmap('gray'))