pyanomaly.datatools.dataclass package

Submodules

pyanomaly.datatools.dataclass.avenue_ped_shanghai module

@author: Yuhao Cheng @contact: yuhao.cheng[at]outlook.com

class pyanomaly.datatools.dataclass.avenue_ped_shanghai.Avenue(frames_folder, clip_length, sampled_clip_length, frame_step=1, clip_step=1, video_format='.mp4', fps=10, transforms=None, is_training=True, one_video=False, only_frame=True, mini=False, extra=False, cfg=None, **kwargs)

Bases: pyanomaly.datatools.abstract.video_dataset.FrameLevelVideoDataset

custom_setup()
class pyanomaly.datatools.dataclass.avenue_ped_shanghai.Ped2(frames_folder, clip_length, sampled_clip_length, frame_step=1, clip_step=1, video_format='.mp4', fps=10, transforms=None, is_training=True, one_video=False, only_frame=True, mini=False, extra=False, cfg=None, **kwargs)

Bases: pyanomaly.datatools.abstract.video_dataset.FrameLevelVideoDataset

custom_setup()
class pyanomaly.datatools.dataclass.avenue_ped_shanghai.Shanghai(frames_folder, clip_length, sampled_clip_length, frame_step=1, clip_step=1, video_format='.mp4', fps=10, transforms=None, is_training=True, one_video=False, only_frame=True, mini=False, extra=False, cfg=None, **kwargs)

Bases: pyanomaly.datatools.abstract.video_dataset.FrameLevelVideoDataset

custom_setup()

pyanomaly.datatools.dataclass.datasets_factory module

@author: Yuhao Cheng @contact: yuhao.cheng[at]outlook.com

class pyanomaly.datatools.dataclass.datasets_factory.VideoAnomalyDatasetFactory(cfg, aug_dict, is_training=True)

Bases: pyanomaly.datatools.abstract.abstract_datasets_factory.AbstractDatasetFactory, pyanomaly.datatools.abstract.abstract_datasets_factory.GetWDataset, pyanomaly.datatools.abstract.abstract_datasets_factory.GetClusterDataset

The factory class to produce the video anomaly dataset class.

NEED_CLUSTER = ['ocae']
NEED_W = ['memae']
NORMAL = ['stae', 'amc', 'anopcn', 'anopred']

Module contents

@author: Yuhao Cheng @contact: yuhao.cheng[at]outlook.com