mwmathis jeylau commited on
Commit
ca3245f
1 Parent(s): 07aa323

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 <jeylau@users.noreply.huggingface.co>

Files changed (3) hide show
  1. README.md +9 -3
  2. detector.pt +3 -0
  3. 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 model is an DLCRNet and/or HRNet-w32 trained on our TopViewMouse-5K dataset.
 
 
 
15
 
16
- • It was trained within the DeepLabCut framework. Full training details can be found in Ye et al. 2023.
 
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("superanimal_topviewmouse", model_dir)
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