• Ting Fu's avatar
    dnn_backend_native_layer_mathunary: add abs support · f73cc61b
    Ting Fu authored
    more math unary operations will be added here
    
    It can be tested with the model file generated with below python scripy:
    
    import tensorflow as tf
    import numpy as np
    import imageio
    
    in_img = imageio.imread('input.jpeg')
    in_img = in_img.astype(np.float32)/255.0
    in_data = in_img[np.newaxis, :]
    
    x = tf.placeholder(tf.float32, shape=[1, None, None, 3], name='dnn_in')
    x1 = tf.subtract(x, 0.5)
    x2 = tf.abs(x1)
    y = tf.identity(x2, name='dnn_out')
    
    sess=tf.Session()
    sess.run(tf.global_variables_initializer())
    
    graph_def = tf.graph_util.convert_variables_to_constants(sess, sess.graph_def, ['dnn_out'])
    tf.train.write_graph(graph_def, '.', 'image_process.pb', as_text=False)
    
    print("image_process.pb generated, please use \
    path_to_ffmpeg/tools/python/convert.py to generate image_process.model\n")
    
    output = sess.run(y, feed_dict={x: in_data})
    imageio.imsave("out.jpg", np.squeeze(output))
    Signed-off-by: 's avatarTing Fu <ting.fu@intel.com>
    Signed-off-by: 's avatarGuo, Yejun <yejun.guo@intel.com>
    f73cc61b
Name
Last commit
Last update
compat Loading commit data...
doc Loading commit data...
ffbuild Loading commit data...
fftools Loading commit data...
libavcodec Loading commit data...
libavdevice Loading commit data...
libavfilter Loading commit data...
libavformat Loading commit data...
libavresample Loading commit data...
libavutil Loading commit data...
libpostproc Loading commit data...
libswresample Loading commit data...
libswscale Loading commit data...
presets Loading commit data...
tests Loading commit data...
tools Loading commit data...
.gitattributes Loading commit data...
.gitignore Loading commit data...
.mailmap Loading commit data...
.travis.yml Loading commit data...
CONTRIBUTING.md Loading commit data...
COPYING.GPLv2 Loading commit data...
COPYING.GPLv3 Loading commit data...
COPYING.LGPLv2.1 Loading commit data...
COPYING.LGPLv3 Loading commit data...
CREDITS Loading commit data...
Changelog Loading commit data...
INSTALL.md Loading commit data...
LICENSE.md Loading commit data...
MAINTAINERS Loading commit data...
Makefile Loading commit data...
README.md Loading commit data...
RELEASE Loading commit data...
configure Loading commit data...