Update README.md
Browse files
README.md
CHANGED
|
@@ -10,13 +10,16 @@ tags:
|
|
| 10 |
|
| 11 |
## Model Details
|
| 12 |
|
| 13 |
-
• SuperAnimal-Quadruped model developed by the [M.W.Mathis Lab](http://www.mackenziemathislab.org/) in 2023, trained to predict quadruped pose from images.
|
| 14 |
Please see [Shaokai Ye et al. 2023](https://arxiv.org/abs/2203.07436) for details.
|
| 15 |
|
| 16 |
-
• The
|
|
|
|
|
|
|
|
|
|
| 17 |
|
| 18 |
-
•
|
| 19 |
-
You can use
|
| 20 |
Here is an example useage:
|
| 21 |
|
| 22 |
```python
|
|
|
|
| 10 |
|
| 11 |
## Model Details
|
| 12 |
|
| 13 |
+
• SuperAnimal-Quadruped model(s) developed by the [M.W.Mathis Lab](http://www.mackenziemathislab.org/) in 2023, trained to predict quadruped pose from images.
|
| 14 |
Please see [Shaokai Ye et al. 2023](https://arxiv.org/abs/2203.07436) for details.
|
| 15 |
|
| 16 |
+
• The there are three models:
|
| 17 |
+
- One is an HRNet-w32 trained within mmpose on our Quadruped-80K dataset.
|
| 18 |
+
- pose_model is an HRNet-w32 compatable for DLC3.0+ Pytorch code, trained on our Quadruped-80K dataset.
|
| 19 |
+
- Detector is a Faster R-CNN that can be used as a detector for top-down estimation.
|
| 20 |
|
| 21 |
+
• Full training details can be found in Ye et al. 2023.
|
| 22 |
+
You can use the pose_model and detector simply with our light-weight loading package called [DLCLibrary](https://github.com/DeepLabCut/DLClibrary).
|
| 23 |
Here is an example useage:
|
| 24 |
|
| 25 |
```python
|