Skip to content

GitLab

  • Menu
Projects Groups Snippets
  • Help
    • Help
    • Support
    • Community forum
    • Submit feedback
    • Contribute to GitLab
  • Sign in
  • F FFmpeg
  • Project information
    • Project information
    • Activity
    • Labels
    • Members
  • Repository
    • Repository
    • Files
    • Commits
    • Branches
    • Tags
    • Contributors
    • Graph
    • Compare
  • Issues 0
    • Issues 0
    • List
    • Boards
    • Service Desk
    • Milestones
  • Merge requests 0
    • Merge requests 0
  • CI/CD
    • CI/CD
    • Pipelines
    • Jobs
    • Schedules
  • Deployments
    • Deployments
    • Environments
    • Releases
  • Monitor
    • Monitor
    • Incidents
  • Packages & Registries
    • Packages & Registries
    • Package Registry
    • Infrastructure Registry
  • Analytics
    • Analytics
    • Value stream
    • CI/CD
    • Repository
  • Wiki
    • Wiki
  • Snippets
    • Snippets
  • Activity
  • Graph
  • Create a new issue
  • Jobs
  • Commits
  • Issue Boards
Collapse sidebar
  • sixue.cheng
  • FFmpeg
  • Repository

Switch branch/tag
  • ffmpeg
  • .gitignore
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: default avatarGuo, Yejun <yejun.guo@intel.com>
    50e194e6

Replace .gitignore

Attach a file by drag & drop or click to upload


Cancel
GitLab will create a branch in your fork and start a merge request.