Image¶
Classes:
|
Metaclass for transformations of images |
|
Do pose estimation with |
-
class
Image
(shape=None, *args, **kwargs)[source]¶ Bases:
autopilot.transform.transforms.Transform
Metaclass for transformations of images
Attributes:
-
property
format_in
¶
-
property
format_out
¶
-
property
shape
¶
-
property
-
class
DLC
(model_dir: Optional[str] = None, model_zoo: Optional[str] = None, *args, **kwargs)[source]¶ Bases:
autopilot.transform.image.Image
Do pose estimation with
DeepLabCut-Live
!!!!!Specify a
model_dir
(relative to<BASEDIR>/dlc
or absolute) or a model name from the DLC model zoo.All other args and kwargs are passed on to
dlclive.DLCLive
, see its documentation for details: https://github.com/DeepLabCut/DeepLabCut-live- Variables
model_type (str, 'local' or 'zoo') – whether a directory (local) or a modelzoo name (zoo) was passed
live (
dlclive.DLCLive
) – the DLCLive object
Must give either
model_dir
ormodel_zoo
- Parameters
model_dir (str) – directory of model, either absolute or relative to
<BASEDIR>/dlc
. ifNone
, usemodel_zoo
model_zoo (str) – name of modelzoo model. if
None
, usemodel_dir
*args – passed to DLCLive and superclass
**kwargs – passed to DLCLive and superclass
Methods:
process
(input)List available modelzoo model names in local deeplabcut version
create_modelzoo
(model)Attributes:
paths used by dlc in manipulating/using models
{prefs.get('BASE_DIR')}/dlc
:returns: str-
process
(input: numpy.ndarray) → numpy.ndarray[source]¶
-
property
model
¶
-
property
model_dir
¶
-
property
dlc_paths
¶ paths used by dlc in manipulating/using models
config: <model_dir>/config.yaml
train_pose_cfg: <model_dir>/dlc-models/iteration-<n>/<name>/train/pose_cfg.yaml,
export_pose_cfg: <model_dir>/exported-models/<name>/pose_cfg.yaml
export_dir: <model_dir>/exported-models/<name>
- Returns
dict
-
property
dlc_dir
¶ {prefs.get('BASE_DIR')}/dlc
:returns: str
-
classmethod
list_modelzoo
()[source]¶ List available modelzoo model names in local deeplabcut version
- Returns
names of available modelzoo models
- Return type
-
property
format_in
¶
-
property
format_out
¶