From fede6ca1dd0077ff509d84bd24028cc7a93bb119 Mon Sep 17 00:00:00 2001 From: StevenLiuWen Date: Tue, 13 Mar 2018 03:28:06 -0400 Subject: first commit --- Codes/flownet2/src/flow_warp.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) create mode 100644 Codes/flownet2/src/flow_warp.py (limited to 'Codes/flownet2/src/flow_warp.py') diff --git a/Codes/flownet2/src/flow_warp.py b/Codes/flownet2/src/flow_warp.py new file mode 100644 index 0000000..fe5fd4d --- /dev/null +++ b/Codes/flownet2/src/flow_warp.py @@ -0,0 +1,15 @@ +import tensorflow as tf + +_flow_warp_ops = tf.load_op_library( + tf.resource_loader.get_path_to_datafile("./ops/build/flow_warp.so")) + + +def flow_warp(image, flow): + return _flow_warp_ops.flow_warp(image, flow) + + +@tf.RegisterGradient("FlowWarp") +def _flow_warp_grad(flow_warp_op, gradients): + return _flow_warp_ops.flow_warp_grad(flow_warp_op.inputs[0], + flow_warp_op.inputs[1], + gradients) -- cgit v1.2.3-70-g09d2