diff options
| author | StevenLiuWen <liuwen@shanghaitech.edu.cn> | 2018-03-13 03:28:06 -0400 |
|---|---|---|
| committer | StevenLiuWen <liuwen@shanghaitech.edu.cn> | 2018-03-13 03:28:06 -0400 |
| commit | fede6ca1dd0077ff509d84bd24028cc7a93bb119 (patch) | |
| tree | af7f6e759b5dec4fc2964daed09e903958b919ed /Codes/flownet2/src/downsample.py | |
first commit
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) |