def grove2():
img1 = open_image(
r"..\..\..\..\resources\eval-twoframes\Grove2\frame10.png")
img2 = open_image(
r"..\..\..\..\resources\eval-twoframes\Grove2\frame11.png")
ref_flow = read_flow_field(
r"..\..\..\..\resources\eval-twoframes-groundtruth\Grove2\flow10.flo")
return img1, img2, ref_flow
def dimetrodon():
img1 = open_image(
r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame10-gray.png")
img2 = open_image(
r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame11-gray.png")
ref_flow = read_flow_field(
r"..\..\..\..\resources\eval-twoframes-groundtruth\Dimetrodon\flow10.flo"
)
return img1, img2, ref_flow
def RubberWhale():
img1 = open_image(
r"..\..\..\..\resources\eval-twoframes\RubberWhale\frame10.png")
img2 = open_image(
r"..\..\..\..\resources\eval-twoframes\RubberWhale\frame11.png")
ref_flow = read_flow_field(
r"..\..\..\..\resources\eval-twoframes-groundtruth\RubberWhale\flow10.flo"
)
return img1, img2, ref_flow
def test_flow():
flow = read_flow_field(
r"..\..\..\..\resources\eval-twoframes-groundtruth\Dimetrodon\flow10.flo"
)
#flow = scale_flow(flow,int(flow.shape[2]/80),int(flow.shape[1]/80))
flow_ = np.array([[[1, 2], [5, 6]], [[8, 7], [4, 3]]])
flow_ = np.full(shape=(2, 4, 4), fill_value=1)
print(flow)
show_flow_field(flow, flow.shape[2], flow.shape[1])
plt.show()
flow_blur = gaussian_blur_flow(flow, 1)
show_flow_field(flow_blur, flow.shape[2], flow.shape[1])
plt.show()
start = time()
plt.figure()
for iter in range(3):
flow = solve_layer(img1,img2,init_flow,settings)
init_flow=flow
return init_flow
if __name__ == '__main__':
#test_setup_linear_system()
img1 = open_image(r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame10.png")
img2 = open_image(r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame11.png")
img1 = open_image(r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame10-gray.png")
img2 = open_image(r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame11-gray.png")
#img1 = open_image(r"..\..\..\..\resources\eval-twoframes\syntetisch\frame10.jpg")
#img2 = open_image(r"..\..\..\..\resources\eval-twoframes\syntetisch\frame11.jpg")
#img1 = img1[[0]]
#img2 = img2[[0]]
#img1 = open_image(r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame10.png")
#img2 = open_image(r"..\..\..\..\resources\eval-twoframes\Dimetrodon\frame11.png")
computed_flow = test_layer1(img1,img2)
ref_flow = read_flow_field(r"..\..\..\..\resources\eval-twoframes-groundtruth\Dimetrodon\flow10.flo")
compare_flow(computed_flow,ref_flow,img1,img2,plot=True)
