Skip to content

Commit 6412489

Browse files
committed
updated README
1 parent da50931 commit 6412489

File tree

1 file changed

+16
-0
lines changed

1 file changed

+16
-0
lines changed

README.md

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,22 @@ model_dir.mkdir()
3535
download_huggingface_model("superanimal_quadruped", model_dir)
3636
```
3737

38+
PyTorch models available for a given dataset (compatible with DeepLabCut>=3.0) can be
39+
listed using the `dlclibrary.get_available_detectors` and
40+
`dlclibrary.get_available_models` methods.
41+
42+
Example use:
43+
44+
```python
45+
>>> import dlclibrary
46+
>>> dlclibrary.get_available_detectors("superanimal_bird")
47+
['fasterrcnn_mobilenet_v3_large_fpn', 'ssdlite']
48+
49+
>>> dlclibrary.get_available_models("superanimal_bird")
50+
['resnet_50']
51+
```
52+
53+
3854
## How to add a new model?
3955

4056
Pick a good model_name. Follow the (novel) naming convention (modeltype_species), e.g. ```superanimal_topviewmouse```.

0 commit comments

Comments
 (0)