Skip to content

  • Projects
  • Groups
  • Snippets
  • Help
    • Loading...
  • Sign in / Register
F
ffmpeg
  • Project
    • Project
    • Details
    • Activity
    • Cycle Analytics
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
    • Charts
  • Issues 0
    • Issues 0
    • List
    • Board
    • Labels
    • Milestones
  • Merge Requests 0
    • Merge Requests 0
  • CI / CD
    • CI / CD
    • Pipelines
    • Jobs
    • Schedules
    • Charts
  • Packages
    • Packages
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Members
    • Members
  • Collapse sidebar
  • Activity
  • Graph
  • Charts
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
  • submodule
  • ffmpeg
  • Repository

Switch branch/tag
  • ffmpeg
  • tools
  • python
  • convert_from_tensorflow.py
Find file
BlameHistoryPermalink
  • Guo, Yejun's avatar
    tools/python: add script to convert TensorFlow model (.pb) to native model (.model) · 50e194e6
    Guo, Yejun authored Jun 13, 2019
    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: 's avatarGuo, Yejun <yejun.guo@intel.com>
    50e194e6
convert_from_tensorflow.py 6.91 KB
EditWeb IDE

Replace convert_from_tensorflow.py

Attach a file by drag & drop or click to upload


Cancel
A new branch will be created in your fork and a new merge request will be started.