diff options
Diffstat (limited to 'Codes/flownet2/src/downsample.py')
| -rw-r--r-- | Codes/flownet2/src/downsample.py | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/Codes/flownet2/src/downsample.py b/Codes/flownet2/src/downsample.py new file mode 100644 index 0000000..5e6fc95 --- /dev/null +++ b/Codes/flownet2/src/downsample.py @@ -0,0 +1,8 @@ +import tensorflow as tf + +_downsample = tf.load_op_library( + tf.resource_loader.get_path_to_datafile("./ops/build/downsample.so")) + + +def downsample(tensor, size): + return _downsample.downsample(tensor, size) |