Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Export module #649

Merged
merged 17 commits into from
Apr 22, 2020
Merged

Export module #649

merged 17 commits into from
Apr 22, 2020

Conversation

gkane26
Copy link
Contributor

@gkane26 gkane26 commented Apr 8, 2020

Adding export module to deeplabcut

changes:

  • adds export.py to deeplabcut/pose_estimation_tensorflow
    • includes export_model function, along with some helpers
      • exports pose_cfg.yaml, checkpoint, and pb files for a specified model
  • adds import for export_model function (from deeplabcut import export_model)
  • adds export_model to cli.py for command line interface to export_model function

To Do

  • currently copies pose_cfg.yaml file for the exported model. Maybe add additional information or change to deploy_cfg or export_cfg?

@AlexEMG AlexEMG self-requested a review April 9, 2020 00:43
@MMathisLab MMathisLab self-requested a review April 9, 2020 15:00
@MMathisLab MMathisLab requested a review from jeylau April 14, 2020 16:59
@MMathisLab
Copy link
Member

perhaps it should specify which snapshot to package, i.e. minimally grab the snapshot from the config.yaml?

@gkane26
Copy link
Contributor Author

gkane26 commented Apr 22, 2020

just added snapshotindex as a parameter to the export_model function and updated export_model in the helper function docs. If provided it will overwrite the snapshotindex in config.yaml. If not provided it will use the specified snapshotindex. If snapshotindex == 'all', it uses snapshotindex = -1

Copy link
Member

@MMathisLab MMathisLab left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

verified functionality.

@MMathisLab MMathisLab added the enhancement New feature or request label Apr 22, 2020
@AlexEMG AlexEMG merged commit 57219eb into DeepLabCut:master Apr 22, 2020
@gkane26 gkane26 deleted the export branch November 6, 2021 14:43
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants