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
|