improved readme, add code to list datasets

Author: n-poulsen

README.md

@@ -36,11 +36,15 @@ download_huggingface_model("superanimal_quadruped", model_dir)
 ```
 
 PyTorch models available for a given dataset (compatible with DeepLabCut>=3.0) can be 
-listed using the `dlclibrary.get_available_detectors` and
-`dlclibrary.get_available_models` methods. Example use:
+listed using the `dlclibrary.get_available_detectors` and 
+`dlclibrary.get_available_models` methods. The datasets for which models are available
+can be listed using `dlclibrary.get_available_datasets`. Example use:
 
 ```python
 >>> import dlclibrary
+>>> dlclibrary.get_available_datasets()
+['superanimal_bird', 'superanimal_topviewmouse', 'superanimal_quadruped']
+
 >>> dlclibrary.get_available_detectors("superanimal_bird")
 ['fasterrcnn_mobilenet_v3_large_fpn', 'ssdlite']
 
@@ -87,4 +91,5 @@ To add a new model for `deeplabcut >= 3.0.0`, simply:
 - add the structure if the model was trained on a new dataset 
 
 The models will then be listed when calling `dlclibrary.get_available_detectors` or
-`dlclibrary.get_available_models`!
+`dlclibrary.get_available_models`! You can list the datasets for which models are 
+available using `dlclibrary.get_available_datasets`.

dlclibrary/__init__.py

@@ -11,6 +11,7 @@
 
 from dlclibrary.dlcmodelzoo.modelzoo_download import (
     download_huggingface_model,
+    get_available_datasets,
     get_available_detectors,
     get_available_models,
     parse_available_supermodels,

dlclibrary/dlcmodelzoo/modelzoo_download.py

@@ -84,6 +84,15 @@ def parse_available_supermodels():
     return super_animal_models
 
 
+def get_available_datasets() -> list[str]:
+    """Only for PyTorch models.
+
+    Returns:
+        The name of datasets for which models are available
+    """
+    return list(_load_pytorch_models().keys())
+
+
 def get_available_detectors(dataset: str) -> list[str]:
     """ Only for PyTorch models.