Skip to content
Permalink
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 @@
"\n",
"This an example notebook for training DeepPoseKit models with DeepLabCut (http://www.deeplabcut.org) formatted data. This notebook shows you how to use the [openfield example dataset](https://github.com/AlexEMG/DeepLabCut/tree/master/examples/openfield-Pranav-2018-10-30) 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",
"\n",
"Be sure to [cite the DeepLabCut toolbox](https://github.com/AlexEMG/DeepLabCut/tree/master/README.md#references) if you use it for your research.\n",
"\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](https://github.com/jgraving/deepposekit/issues/new) or [submitting a pull request](https://help.github.com/en/articles/creating-a-pull-request-from-a-fork)\n",
"\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",
"\n",
"#model = LEAP(train_generator)\n",
"#model = StackedHourglass(train_generator)\n",
"\n",
"model.get_config()"
]
},

0 comments on commit f5313e2

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