fastvideo.v1.dataset.t2v_datasets

`fastvideo.v1.dataset.t2v_datasets`#

Module Contents#

Classes#

`DataSetProg`
`SingletonMeta`
`T2V_dataset`

Functions#

filter_resolution

Data#

dataset_prog

API#

class fastvideo.v1.dataset.t2v_datasets.DataSetProg[source]#

Initialization

get_item(work_info) → int[source]#

set_cap_list(num_workers, cap_list, n_elements) → None[source]#

class fastvideo.v1.dataset.t2v_datasets.SingletonMeta[source]#: Bases: type

class fastvideo.v1.dataset.t2v_datasets.T2V_dataset(args, transform, temporal_sample, tokenizer, transform_topcrop, start_idx=0)[source]#

Bases: torch.utils.data.Dataset

decord_read(path, frame_indices) → torch.Tensor[source]#

define_frame_index(cap_list) → tuple[list[dict], list[int]][source]#

get_cap_list() → list[source]#

get_data(idx) → dict[source]#

get_image(idx) → dict[source]#

get_video(idx) → dict[source]#

read_jsons(data) → list[dict][source]#

set_checkpoint(n_used_elements)[source]#

fastvideo.v1.dataset.t2v_datasets.dataset_prog[source]#: ‘DataSetProg(…)’

fastvideo.v1.dataset.t2v_datasets.filter_resolution(h: int, w: int, max_h_div_w_ratio: float = 17 / 16, min_h_div_w_ratio: float = 8 / 16) → bool[source]#