mirror of
https://git.ffmpeg.org/ffmpeg.git
synced 2024-12-24 16:22:37 +00:00
c0cdeea0ee
It can be tested with the model generated with below python script: 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') please uncomment the part you want to test x_sinh_1 = tf.sinh(x) x_out = tf.divide(x_sinh_1, 1.176) # sinh(1.0) x_cosh_1 = tf.cosh(x) x_out = tf.divide(x_cosh_1, 1.55) # cosh(1.0) x_tanh_1 = tf.tanh(x) x__out = tf.divide(x_tanh_1, 0.77) # tanh(1.0) x_asinh_1 = tf.asinh(x) x_out = tf.divide(x_asinh_1, 0.89) # asinh(1.0/1.1) x_acosh_1 = tf.add(x, 1.1) x_acosh_2 = tf.acosh(x_acosh_1) # accept (1, inf) x_out = tf.divide(x_acosh_2, 1.4) # acosh(2.1) x_atanh_1 = tf.divide(x, 1.1) x_atanh_2 = tf.atanh(x_atanh_1) # accept (-1, 1) x_out = tf.divide(x_atanh_2, 1.55) # atanhh(1.0/1.1) y = tf.identity(x_out, name='dnn_out') #please only preserve the x_out you want to test 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: Ting Fu <ting.fu@intel.com> |
||
---|---|---|
.. | ||
python | ||
.gitignore | ||
aviocat.c | ||
bisect-create | ||
bookmarklets.html | ||
cl2c | ||
clean-diff | ||
coverity.c | ||
crypto_bench.c | ||
cws2fws.c | ||
dvd2concat | ||
enum_options.c | ||
ffescape.c | ||
ffeval.c | ||
ffhash.c | ||
fourcc2pixfmt.c | ||
gen-rc | ||
graph2dot.c | ||
ismindex.c | ||
libav-merge-next-commit | ||
loudnorm.rb | ||
make_chlayout_test | ||
Makefile | ||
missing_codec_desc | ||
murge | ||
normalize.py | ||
patcheck | ||
pktdumper.c | ||
plotframes | ||
probetest.c | ||
qt-faststart.c | ||
seek_print.c | ||
sidxindex.c | ||
sofa2wavs.c | ||
target_bsf_fuzzer.c | ||
target_dec_fate.list | ||
target_dec_fate.sh | ||
target_dec_fuzzer.c | ||
target_dem_fuzzer.c | ||
trasher.c | ||
uncoded_frame.c | ||
unwrap-diff | ||
venc_data_dump.c | ||
yuvcmp.c | ||
zmqsend.c | ||
zmqshell.py |