mirror of https://git.ffmpeg.org/ffmpeg.git
fab00b0ae0
It can be tested with the model generated with below python script:
import tensorflow as tf
import os
import numpy as np
import imageio
from tensorflow.python.framework import graph_util
name = 'floor'
pb_file_path = os.getcwd()
if not os.path.exists(pb_file_path+'/{}_savemodel/'.format(name)):
os.mkdir(pb_file_path+'/{}_savemodel/'.format(name))
with tf.Session(graph=tf.Graph()) as sess:
in_img = imageio.imread('detection.jpg')
in_img = in_img.astype(np.float32)
in_data = in_img[np.newaxis, :]
input_x = tf.placeholder(tf.float32, shape=[1, None, None, 3], name='dnn_in')
y_ = tf.math.floor(input_x*255)/255
y = tf.identity(y_, name='dnn_out')
sess.run(tf.global_variables_initializer())
constant_graph = graph_util.convert_variables_to_constants(sess, sess.graph_def, ['dnn_out'])
with tf.gfile.FastGFile(pb_file_path+'/{}_savemodel/model.pb'.format(name), mode='wb') as f:
f.write(constant_graph.SerializeToString())
print("model.pb generated, please in ffmpeg path use\n \n \
python tools/python/convert.py {}_savemodel/model.pb --outdir={}_savemodel/ \n \nto generate model.model\n".format(name,name))
output = sess.run(y, feed_dict={ input_x: in_data})
imageio.imsave("out.jpg", np.squeeze(output))
print("To verify, please ffmpeg path use\n \n \
./ffmpeg -i detection.jpg -vf format=rgb24,dnn_processing=model={}_savemodel/model.pb:input=dnn_in:output=dnn_out:dnn_backend=tensorflow -f framemd5 {}_savemodel/tensorflow_out.md5\n \
or\n \
./ffmpeg -i detection.jpg -vf format=rgb24,dnn_processing=model={}_savemodel/model.pb:input=dnn_in:output=dnn_out:dnn_backend=tensorflow {}_savemodel/out_tensorflow.jpg\n \nto generate output result of tensorflow model\n".format(name, name, name, name))
print("To verify, please ffmpeg path use\n \n \
./ffmpeg -i detection.jpg -vf format=rgb24,dnn_processing=model={}_savemodel/model.model:input=dnn_in:output=dnn_out:dnn_backend=native -f framemd5 {}_savemodel/native_out.md5\n \
or \n \
./ffmpeg -i detection.jpg -vf format=rgb24,dnn_processing=model={}_savemodel/model.model:input=dnn_in:output=dnn_out:dnn_backend=native {}_savemodel/out_native.jpg\n \nto generate output result of native model\n".format(name, name, name, name))
Signed-off-by: Mingyu Yin <mingyu.yin@intel.com>
|
||
|---|---|---|
| .. | ||
| python | ||
| .gitignore | ||
| Makefile | ||
| 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 | ||
| 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 | ||