pulsatrix
Loading...
Searching...
No Matches
video_transforms.hpp
Go to the documentation of this file.
1
5#pragma once
6
9
10namespace pulsatrix {
11
24public:
25 UniformFrameSampleTransform(int64_t num_frames, DeviceBackend* backend)
26 : num_frames_(num_frames), backend_(backend) {}
27
32 [[nodiscard]] Sample apply(Sample sample) const override;
33
34private:
35 int64_t num_frames_;
36 DeviceBackend* backend_;
37};
38
39} // namespace pulsatrix
Vendor-agnostic compute/memory backend. CPUBackend, CUDABackend (Phase 1.5), and HIPBackend (Phase 1....
Definition device_backend.hpp:219
A single sample-level preprocessing step (normalize, augment, tokenize, ...).
Definition transform.hpp:20
Samples num_frames evenly-spaced frames from a clip Tensor (sample.fields[0], shape (T,...
Definition video_transforms.hpp:23
Sample apply(Sample sample) const override
UniformFrameSampleTransform(int64_t num_frames, DeviceBackend *backend)
Definition video_transforms.hpp:25
Abstract interface isolating vendor-specific memory/compute operations from Tensor/ComputationGraph.
Definition acquisition_functions.hpp:16
One dataset sample: an ordered list of Tensor fields (e.g. {features, label} or {image,...
Definition dataset.hpp:19
Sample-level preprocessing abstraction – Transform, Compose, TransformDataset.