Visitar URL original
DeepLabCut 3.0 - Updates to COLABs and Notebooks for the PyTorch Engine by n-poulsen · Pull Request #2868 · DeepLabCut/DeepLabCut · GitHub
Skip to content

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

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.

Sign up for GitHub

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

Merged
MMathisLab merged 18 commits into from
Feb 28, 2025
Merged

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

Changes from 1 commit
Commits
Show all changes
18 commits
Select commit Hold shift + click to select a range
ba8569d
update openfield COLAB
Feb 7, 2025
a079aa5
updated COLAB_YOURDATA_TrainNetwork_VideoAnalysis and COLAB_maDLC_Tra…
Feb 7, 2025
f0c519c
fix ReID notebook
Feb 7, 2025
f67f24d
fix ReID notebook
Feb 7, 2025
6974baa
save predictions to destfolder
Feb 7, 2025
8a53a06
update superanimal notebook
Feb 7, 2025
3c3ca56
Merge branch 'pytorch_dlc' into niels/colabs_and_notebooks
Feb 7, 2025
9ac9290
fixed demo_data.py, Demo_labeledexample_MouseReaching.ipynb
Feb 7, 2025
db7c750
set default batch size to 8, fix notebooks
Feb 7, 2025
6e37654
updated notebooks
Feb 7, 2025
c2c9a81
Merge branch 'pytorch_dlc' into niels/colabs_and_notebooks
n-poulsen Feb 7, 2025
bf0ff83
fix edited config
Feb 7, 2025
ee50925
fix edited config
Feb 7, 2025
d9a669f
fix edited config
Feb 7, 2025
e3c40e4
Merge branch 'pytorch_dlc' into niels/colabs_and_notebooks
Feb 14, 2025
9412886
update 3mice demo
Feb 14, 2025
c6edd9e
install deeplabcut through prerelease
Feb 28, 2025
c9ff454
Merge branch 'pytorch_dlc' into niels/colabs_and_notebooks
MMathisLab Feb 28, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update superanimal notebook
  • Loading branch information
Niels Poulsen
Niels Poulsen committed Feb 7, 2025
commit 8a53a06cbed64d74a2e654b3661de69d0f8a9ea0
22 changes: 9 additions & 13 deletions examples/COLAB/COLAB_YOURDATA_SuperAnimal.ipynb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -43,9 +43,7 @@
"outputId": "290a589f-a063-4933-d315-e13052ec1024"
},
"outputs": [],
"source": [
"!pip install \"git+https://github.com/DeepLabCut/DeepLabCut.git@pytorch_dlc#egg=deeplabcut[modelzoo]\""
]
"source": "!pip install \"deeplabcut>=3\""
},
{
"cell_type": "markdown",
Expand Down Expand Up @@ -83,8 +81,6 @@
"from deeplabcut.pose_estimation_pytorch.apis import (\n",
" superanimal_analyze_images,\n",
")\n",
"from deeplabcut.core.weight_init import WeightInitialization\n",
"from deeplabcut.core.engine import Engine\n",
"from deeplabcut.modelzoo import build_weight_init\n",
"from deeplabcut.modelzoo.utils import (\n",
" create_conversion_table,\n",
Expand Down Expand Up @@ -541,7 +537,7 @@
" Shuffles=[imagenet_transfer_learning_shuffle],\n",
" net_type=f\"top_down_{model_name}\",\n",
" detector_type=detector_name,\n",
" engine=Engine.PYTORCH,\n",
" engine=deeplabcut.Engine.PYTORCH,\n",
" userfeedback=False,\n",
")"
]
Expand Down Expand Up @@ -600,7 +596,7 @@
" epochs=50,\n",
" save_epochs=10,\n",
" batch_size=64, # if you get a CUDA OOM error when training on a GPU, reduce to 32, 16, ...!\n",
" display_iters=10,\n",
" displayiters=10,\n",
" shuffle=imagenet_transfer_learning_shuffle,\n",
")"
]
Expand Down Expand Up @@ -673,7 +669,7 @@
" config_path,\n",
" from_shuffle=imagenet_transfer_learning_shuffle,\n",
" shuffles=[superanimal_transfer_learning_shuffle],\n",
" engine=Engine.PYTORCH,\n",
" engine=deeplabcut.Engine.PYTORCH,\n",
" net_type=f\"top_down_{model_name}\",\n",
" detector_type=detector_name,\n",
" weight_init=weight_init,\n",
Expand Down Expand Up @@ -736,7 +732,7 @@
" epochs=50,\n",
" save_epochs=10,\n",
" batch_size=64, # if you get a CUDA OOM error when training on a GPU, reduce to 32, 16, ...!\n",
" display_iters=10,\n",
" displayiters=10,\n",
" shuffle=superanimal_transfer_learning_shuffle,\n",
")"
]
Expand Down Expand Up @@ -959,7 +955,7 @@
" config_path,\n",
" from_shuffle=imagenet_transfer_learning_shuffle,\n",
" shuffles=[superanimal_naive_finetune_shuffle],\n",
" engine=Engine.PYTORCH,\n",
" engine=deeplabcut.Engine.PYTORCH,\n",
" net_type=f\"top_down_{model_name}\",\n",
" detector_type=detector_name,\n",
" weight_init=weight_init,\n",
Expand Down Expand Up @@ -993,7 +989,7 @@
" epochs=50,\n",
" save_epochs=10,\n",
" batch_size=64, # if you get a CUDA OOM error when training on a GPU, reduce to 32, 16, ...!\n",
" display_iters=10,\n",
" displayiters=10,\n",
" shuffle=superanimal_naive_finetune_shuffle,\n",
")"
]
Expand Down Expand Up @@ -1092,7 +1088,7 @@
" config_path,\n",
" from_shuffle=imagenet_transfer_learning_shuffle,\n",
" shuffles=[superanimal_memory_replay_shuffle],\n",
" engine=Engine.PYTORCH,\n",
" engine=deeplabcut.Engine.PYTORCH,\n",
" net_type=f\"top_down_{model_name}\",\n",
" detector_type=detector_name,\n",
" weight_init=weight_init,\n",
Expand Down Expand Up @@ -1126,7 +1122,7 @@
" epochs=50,\n",
" save_epochs=10,\n",
" batch_size=64, # if you get a CUDA OOM error when training on a GPU, reduce to 32, 16, ...!\n",
" display_iters=10,\n",
" displayiters=10,\n",
" shuffle=superanimal_memory_replay_shuffle,\n",
")"
]
Expand Down