FazBrowse GitHub Viewer | Trending |
URL:
| Home
Tools: [Download Repo ZIP]   [Original HTTPS Page]

DLC 3.0 - Function to export PyTorch models by n-poulsen · Pull Request #2800 · DeepLabCut/DeepLabCut · GitHub

/ DeepLabCut Public

DLC 3.0 - Function to export PyTorch models - #2800

Merged
MMathisLab merged 16 commits into
pytorch_dlcfrom
niels/torch_export_model
Dec 3, 2024
ConversationCommits (16)ChecksFiles changed
Merged

DLC 3.0 - Function to export PyTorch models#2800
MMathisLab merged 16 commits into
pytorch_dlcfrom
niels/torch_export_model

Conversation

n-poulsen commented Dec 3, 2024

Copy link
Copy Markdown
Contributor

Implements the export_model API method for models trained with the PyTorch engine. Tests are added for the different features of the export_model method.

The following modifications are made to implement the feature:

  • The DLCLoader lazy-loads the DataFrames and train/test splits
  • We can instantiate the DLCLoader with a project configuration directly (a dict) instead of having to pass a filepath
    • This makes it possible to change the project iteration dynamically when exporting models

n-poulsen added enhancement New feature or request DLC3.0🔥 labels Dec 3, 2024
n-poulsen requested a review from maximpavliv December 3, 2024 11:34
maximpavliv approved these changes Dec 3, 2024
View reviewed changes

maximpavliv left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low Quality

LGTM!

MMathisLab merged commit 5b6c8b7 into pytorch_dlc Dec 3, 2024
MMathisLab deleted the niels/torch_export_model branch April 15, 2025 11:36

hg-aus-g commented Apr 24, 2025 •
edited
Loading

Copy link
Copy Markdown

Does the export_model() function works with pytorch models? Unfortunately I got only one .pt file. But I need .pb file for use witch dlc-live-gui!

Thanks!

maximpavliv commented May 5, 2025

Copy link
Copy Markdown
Contributor

Hello @hg-aus-g ,

.pb is the format used by Tensorflow, Pytorch uses the .pt format to save models, so this is perfectly normal.

The thing is that pytorch support has not been merged on the DeepLabCut-live side yet, although it is almost ready (see PR). It should be merged very soon :)

This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters. Learn more about bidirectional Unicode characters
Show hidden characters
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

1 more reviewer
maximpavliv maximpavliv approved these changes
Reviewers whose approvals may not affect merge requirements

Assignees

No one assigned

Labels

DLC3.0🔥 enhancement New feature or request

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

4 participants

Back | FazBrowse Home | New Git URL