ffmpeg/tools
Guo, Yejun 50e194e6e1 tools/python: add script to convert TensorFlow model (.pb) to native model (.model)
For example, given TensorFlow model file espcn.pb,
to generate native model file espcn.model, just run:
python convert.py espcn.pb

In current implementation, the native model file is generated for
specific dnn network with hard-code python scripts maintained out of ffmpeg.
For example, srcnn network used by vf_sr is generated with
https://github.com/HighVoltageRocknRoll/sr/blob/master/generate_header_and_model.py#L85

In this patch, the script is designed as a general solution which
converts general TensorFlow model .pb file into .model file. The script
now has some tricky to be compatible with current implemention, will
be refined step by step.

The script is also added into ffmpeg source tree. It is expected there
will be many more patches and community needs the ownership of it.

Another technical direction is to do the conversion in c/c++ code within
ffmpeg source tree. While .pb file is organized with protocol buffers,
it is not easy to do such work with tiny c/c++ code, see more discussion
at http://ffmpeg.org/pipermail/ffmpeg-devel/2019-May/244496.html. So,
choose the python script.

Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
2019-07-01 10:23:47 -03:00
..
python tools/python: add script to convert TensorFlow model (.pb) to native model (.model) 2019-07-01 10:23:47 -03:00
.gitignore
Makefile tools: Add fuzzer for demuxers 2019-05-31 22:44:30 +02:00
aviocat.c
bisect-create
bookmarklets.html
cl2c
clean-diff
coverity.c
crypto_bench.c tools/crypto_bench: update the comment about build command 2019-05-09 17:03:30 +08:00
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_dec_fate.list tools/target_dec_fate.list: add issues 4000 to 6000 2019-03-20 21:12:45 +01:00
target_dec_fate.sh tools/target_dec_fate.sh: Add support for lines that are comments 2019-03-01 23:25:08 +01:00
target_dec_fuzzer.c tools/target_dec_fuzzer: Also fuzz extradata 2019-06-14 21:36:39 +02:00
target_dem_fuzzer.c tools: Add fuzzer for demuxers 2019-05-31 22:44:30 +02:00
trasher.c
uncoded_frame.c
unwrap-diff
yuvcmp.c
zmqsend.c
zmqshell.py