def main():
if len(sys.argv) < 3:
print('\tUSAGE: compute_flow_directory.py [PARAMETER] INDIR OUTDIR')
print('\t\tSee readme.md for details.')
sys.exit(1)
if '-kitti' in sys.argv:
use_dataset = 'kitti'
else:
use_dataset = 'sintel'
if '-pcaflow' in sys.argv:
use_algorithm = 'pcaflow'
else:
use_algorithm = 'pcalayers'
preset = '{}_{}'.format(use_algorithm,use_dataset)
outdir = sys.argv[-1]
indir = sys.argv[-2]
if not os.path.isdir(outdir):
os.makedirs(outdir)
PATH_PC_U = 'data/PC_U.npy'
PATH_PC_V = 'data/PC_V.npy'
if use_dataset == 'sintel':
PATH_COV = 'data/COV_SINTEL.npy'
PATH_COV_SUBLAYER = 'data/COV_SINTEL_SUBLAYER.npy'
else:
PATH_COV = 'data/COV_KITTI.npy'
PATH_COV_SUBLAYER = None
P = PCAFlow.PCAFlow(
pc_file_u=PATH_PC_U,
pc_file_v=PATH_PC_V,
covfile=PATH_COV,
covfile_sublayer=PATH_COV_SUBLAYER,
preset=preset,
)
files_input = [f for f in sorted(os.listdir(indir)) if f.endswith('.png')]
print('Number of input files: {}'.format(len(files_input)))
for i,fname in enumerate(files_input[:-1]):
fullpath = lambda x : os.path.join(indir,x)
if i==0:
P.push_back(imread(fullpath(fname)))
print('Adding {}'.format(fname))
P.push_back(imread(fullpath(files_input[i+1])))
print('Adding {}'.format(files_input[i+1]))
u,v = P.compute_flow()
outfile = os.path.join(outdir, fname[:-3] + 'flo')
# Save output files
I_flow = viz_flow(u,v)
flow_write(outfile,u,v)
imsave(outfile + '.png', I_flow)
def main():
if len(sys.argv) < 4:
print('\tUSAGE: compute_flow.py [PARAMETER] IMAGE1 IMAGE2 OUTFILE.')
print('\t\tSee readme.md for details.')
sys.exit(1)
if '-kitti' in sys.argv:
use_dataset = 'kitti'
else:
use_dataset = 'sintel'
if '-pcaflow' in sys.argv:
use_algorithm = 'pcaflow'
else:
use_algorithm = 'pcalayers'
preset = '{}_{}'.format(use_algorithm,use_dataset)
outfile = sys.argv[-1]
img2 = sys.argv[-2]
img1 = sys.argv[-3]
PATH_PC_U = 'data/PC_U.npy'
PATH_PC_V = 'data/PC_V.npy'
if use_dataset == 'sintel':
PATH_COV = 'data/COV_SINTEL.npy'
PATH_COV_SUBLAYER = 'data/COV_SINTEL_SUBLAYER.npy'
else:
PATH_COV = 'data/COV_KITTI.npy'
PATH_COV_SUBLAYER = None
P = PCAFlow.PCAFlow(
pc_file_u=PATH_PC_U,
pc_file_v=PATH_PC_V,
covfile=PATH_COV,
covfile_sublayer=PATH_COV_SUBLAYER,
preset=preset,
)
I1 = imread(img1)
I2 = imread(img2)
# Compute flow
P.push_back(I1)
P.push_back(I2)
u,v = P.compute_flow()
# Save output files
I_flow = viz_flow(u,v)
flow_write(outfile,u,v)
imsave(outfile + '.png', I_flow)
def main():
if len(sys.argv) < 4:
print('\tUSAGE: compute_flow.py [PARAMETER] IMAGE1 IMAGE2 OUTFILE.')
print('\t\tSee readme.md for details.')
sys.exit(1)
if '-kitti' in sys.argv:
use_dataset = 'kitti'
else:
use_dataset = 'sintel'
if '-pcaflow' in sys.argv:
use_algorithm = 'pcaflow'
else:
use_algorithm = 'pcalayers'
preset = '{}_{}'.format(use_algorithm, use_dataset)
outfile = sys.argv[-1]
img2 = sys.argv[-2]
img1 = sys.argv[-3]
PATH_PC_U = 'data/PC_U.npy'
PATH_PC_V = 'data/PC_V.npy'
if use_dataset == 'sintel':
PATH_COV = 'data/COV_SINTEL.npy'
PATH_COV_SUBLAYER = 'data/COV_SINTEL_SUBLAYER.npy'
else:
PATH_COV = 'data/COV_KITTI.npy'
PATH_COV_SUBLAYER = None
P = PCAFlow.PCAFlow(
pc_file_u=PATH_PC_U,
pc_file_v=PATH_PC_V,
covfile=PATH_COV,
covfile_sublayer=PATH_COV_SUBLAYER,
preset=preset,
)
I1 = imread(img1)
I2 = imread(img2)
# Compute flow
P.push_back(I1)
P.push_back(I2)
u, v = P.compute_flow()
# Save output files
I_flow = viz_flow(u, v)
flow_write(outfile, u, v)
imsave(outfile + '.png', I_flow)
### Once the object is created, it can be used like this:
I1 = imread('image1.png')
I2 = imread('image2.png')
P.push_back(I1)
P.push_back(I2)
# Compute flow
u,v = P.compute_flow()
### Use this if you want to just get the motion descriptor
#u,v,data = P.compute_flow(return_additional=['weights',])
#descriptor = data['weights']
I_flow = viz_flow(u,v)
plt.figure()
plt.subplot(221)
plt.imshow(I1)
plt.title('First image')
plt.subplot(222)
plt.imshow(I_flow)
plt.title('Flow colormap')
plt.subplot(223)
plt.imshow(u)
plt.title('Horizontal component')
plt.subplot(224)
plt.imshow(v)
plt.title('Vertical component')
