Skip to content

DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine#2868

Merged
MMathisLab merged 18 commits into
pytorch_dlcfrom
niels/colabs_and_notebooks
Feb 28, 2025
Merged

DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine#2868
MMathisLab merged 18 commits into
pytorch_dlcfrom
niels/colabs_and_notebooks

Conversation

@n-poulsen

@n-poulsen n-poulsen commented Feb 7, 2025

Copy link
Copy Markdown
Contributor

This pull request:

  • Updates all DeepLabCut example Jupyter Notebooks for use with the PyTorch engine
  • Updates all COLAB Notebooks for use with the PyTorch engine
    • Installation of DeepLabCut assumes the pytorch_dlc branch was merged into main or a 3.0 release of DeepLabCut was made to pypi (depending on the notebook). Until that's the case, !pip install "deeplabcut>=3" need to be replaced with !pip install "git+https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc#egg=deeplabcut and !git checkout pytorch_dlc needs to be run in COLABs where the DeepLabCut repo is cloned.
    • The COLAB_3miceDemo.ipynb notebook is updated to point to other notebooks and keeps its TensorFlow engine
    • COLAB_maDLC_TrainNetwork_VideoAnalysis.ipynb is renamed to COLAB_YOURDATA_maDLC_TrainNetwork_VideoAnalysis.ipynb as it is intended to be used with users' data
  • Updates the deeplabcut/create_project/demo_data.py script to set the Engine to PyTorch

Notebooks were tested locally, and COLABs were tested on Colab.

@n-poulsen n-poulsen added howtousedlc issue regarding the use of DeepLabCut COLAB Jupyter related to jupyter notebooks DLC3.0🔥 labels Feb 7, 2025
@n-poulsen n-poulsen requested a review from maximpavliv February 7, 2025 15:28
@n-poulsen n-poulsen changed the title DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine [WIP] DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine Feb 7, 2025

@maximpavliv 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.

Niice, thanks @n-poulsen !!

@n-poulsen n-poulsen changed the title [WIP] DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine Feb 28, 2025
@MMathisLab MMathisLab merged commit 7b43b35 into pytorch_dlc Feb 28, 2025
@MMathisLab MMathisLab deleted the niels/colabs_and_notebooks branch February 28, 2025 17:30
@deruyter92 deruyter92 mentioned this pull request May 21, 2026
1 task
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

COLAB DLC3.0🔥 howtousedlc issue regarding the use of DeepLabCut Jupyter related to jupyter notebooks

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants