Skip to content
Browse files

Update example notebooks

  • Loading branch information...
jgraving committed Oct 10, 2019
1 parent 5994662 commit f5313e22db6122b0b855297ba59a309618ccd579
Showing with 5 additions and 1 deletion.
  1. +2 −0 examples/deeplabcut_data_example.ipynb
  2. +3 −1 examples/step3_train_model.ipynb
@@ -15,6 +15,8 @@
"This an example notebook for training DeepPoseKit models with DeepLabCut ( formatted data. This notebook shows you how to use the [openfield example dataset]( from Mathis et al. (2018) to train a deep learning model using data augmentation and callbacks for logging the training process and saving the best model during training.\n",
"Be sure to [cite the DeepLabCut toolbox]( if you use it for your research.\n",
"**NOTE**: This example currently only supports basic functionality from DeepLabCut (no cropping, etc.). It assumes each image contains a single individual (or multiple individuals are labeled as separate keypoints) and assumes all images are the same size/resolution. If you run into problems, you can help us improve DeepPoseKit by [opening an issue]( or [submitting a pull request](\n",
"**If you're using Colab**: make sure to go to the “Runtime” dropdown menu, select “Change runtime type” and select `GPU` in the \"Hardware accelerator\" drop-down menu\n"
@@ -441,7 +441,7 @@
"metadata": {},
"outputs": [],
"source": [
"from deepposekit.models import DeepLabCut, StackedDenseNet, LEAP"
"from deepposekit.models import DeepLabCut, StackedDenseNet, StackedHourglass, LEAP"
@@ -482,6 +482,8 @@
"#model = DeepLabCut(train_generator, backbone=\"densenet121\")\n",
"#model = LEAP(train_generator)\n",
"#model = StackedHourglass(train_generator)\n",

0 comments on commit f5313e2

Please sign in to comment.
You can’t perform that action at this time.