Add DLC3 superanimal_topviewmouse detector and pose model (#1)
Browse files- Add DLC3 superanimal_topviewmouse detector and pose model (c600ad6c5bbd0e9ce58cec6da7f4f3f1880f961e)
- Update README.md (159c9787a3b1e3e03960d2db07274ecd4f8fa1b0)
Co-authored-by: Jessy Lauer <[email protected]>
- README.md +9 -3
- detector.pt +3 -0
- pose_model.pth +3 -0
README.md
CHANGED
|
@@ -2,6 +2,8 @@
|
|
| 2 |
tags:
|
| 3 |
- computer_vision
|
| 4 |
- pose_estimation
|
|
|
|
|
|
|
| 5 |
---
|
| 6 |
# MODEL CARD:
|
| 7 |
|
|
@@ -11,9 +13,13 @@ tags:
|
|
| 11 |
trained to predict mouse topline pose from images.
|
| 12 |
Please see [Shaokai Ye et al. 2023](https://arxiv.org/abs/2203.07436) for details.
|
| 13 |
|
| 14 |
-
• The
|
|
|
|
|
|
|
|
|
|
| 15 |
|
| 16 |
-
|
|
|
|
| 17 |
You can use this model simply with our light-weight loading package called [DLCLibrary](https://github.com/DeepLabCut/DLClibrary).
|
| 18 |
Here is an example useage:
|
| 19 |
|
|
@@ -24,7 +30,7 @@ from dlclibrary import download_huggingface_model
|
|
| 24 |
# Creates a folder and downloads the model to it
|
| 25 |
model_dir = Path("./superanimal_topviewmouse_model")
|
| 26 |
model_dir.mkdir()
|
| 27 |
-
download_huggingface_model("
|
| 28 |
```
|
| 29 |
|
| 30 |
## Intended Use
|
|
|
|
| 2 |
tags:
|
| 3 |
- computer_vision
|
| 4 |
- pose_estimation
|
| 5 |
+
- animal_pose_estimation
|
| 6 |
+
- deeplabcut
|
| 7 |
---
|
| 8 |
# MODEL CARD:
|
| 9 |
|
|
|
|
| 13 |
trained to predict mouse topline 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 |
+
- `pose_model.pth` is an HRNet-w32 compatable for DLC3.0+ Pytorch code, trained on our TopViewMouse-5K dataset.
|
| 18 |
+
- `detector.pt` is a Faster R-CNN that can be used as a detector for top-down detection.
|
| 19 |
+
- `DLC_ma_supertopview5k_resnet_50_iteration-0_shuffle-1.tar.gz` is a DLCRNet trained on our TopViewMouse-5K dataset.
|
| 20 |
|
| 21 |
+
|
| 22 |
+
• Full training details can be found in Ye et al. 2023.
|
| 23 |
You can use this model simply with our light-weight loading package called [DLCLibrary](https://github.com/DeepLabCut/DLClibrary).
|
| 24 |
Here is an example useage:
|
| 25 |
|
|
|
|
| 30 |
# Creates a folder and downloads the model to it
|
| 31 |
model_dir = Path("./superanimal_topviewmouse_model")
|
| 32 |
model_dir.mkdir()
|
| 33 |
+
download_huggingface_model("superanimal_topviewmouse_model", model_dir)
|
| 34 |
```
|
| 35 |
|
| 36 |
## Intended Use
|
detector.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:5108318a1e63dcc4103de29b98a94c5a42302e6104637a38dacf0fef0f137dea
|
| 3 |
+
size 517816013
|
pose_model.pth
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7ec111e8d9a22277e3b841ecafc6159b1c6501ab7fd4af92cf60b68a7f12fdd7
|
| 3 |
+
size 159594859
|