Skip to content

Conversation

@n-poulsen
Copy link
Contributor

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

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
Copy link
Contributor

@maximpavliv maximpavliv left a comment

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
1 check was pending
@MMathisLab MMathisLab deleted the niels/colabs_and_notebooks branch February 28, 2025 17:30
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.

4 participants