Add DLC3 superanimal_topviewmouse detector and pose model
#1
by
jeylau
- opened
- 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
|