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:
- 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- property dlc_paths: dict
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