File tree Expand file tree Collapse file tree 1 file changed +16
-0
lines changed
Expand file tree Collapse file tree 1 file changed +16
-0
lines changed Original file line number Diff line number Diff line change @@ -35,6 +35,22 @@ model_dir.mkdir()
3535download_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
4056Pick a good model_name. Follow the (novel) naming convention (modeltype_species), e.g. ``` superanimal_topviewmouse ``` .
You can’t perform that action at this time.
0 commit comments