/
tutorial-part2small.py
56 lines (39 loc) · 1.51 KB
/
tutorial-part2small.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
51
52
53
54
55
56
#!/usr/bin/env python
import sys
if 'OpenPIV' not in sys.path:
sys.path.append('/Users/alex/Documents/OpenPIV/alexlib/openpiv-python')
from openpiv import tools, validation, process, filters, scaling, pyprocess
import numpy as np
import matplotlib.pyplot as plt
frame_a = tools.imread('exp1_001_a.bmp')
frame_b = tools.imread('exp1_001_b.bmp' )
frame_a = frame_a[:128,:128]
frame_b = frame_b[:128,:128]
from pylab import *
imshow(np.c_[frame_a,frame_b],cmap=cm.gray)
show()
u, v, sig2noise = process.extended_search_area_piv(
frame_a.astype(np.int32),
frame_b.astype(np.int32),
window_size=24, overlap=12,
dt=0.02, search_area_size=32,
sig2noise_method='peak2peak' )
x, y = process.get_coordinates( image_size=frame_a.shape, window_size=24, overlap=12 )
plt.figure()
plt.quiver(x,y,u,v)
plt.show()
u, v, s2n = pyprocess.piv( frame_a, frame_b, corr_method='fft',
window_size=32, overlap=12, dt=0.02, sig2noise_method='peak2peak' )
x, y = pyprocess.get_coordinates( image_size=frame_a.shape, window_size=32, overlap=12 )
plt.figure()
plt.quiver(x,y,u,v)
plt.show()
u, v, s2n = pyprocess.piv( frame_a, frame_b, corr_method='direct',
window_size=24, overlap=12, dt=0.02,
sig2noise_method='peak2peak',
search_size = 32)
x, y = pyprocess.get_coordinates( image_size=frame_a.shape,
window_size=24, overlap=12 )
plt.figure()
plt.quiver(x,y,u,v)
plt.show()