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)