stats

type EpisodeInput ¶

type EpisodeInput struct {
	Columns    map[string]any
	FramePaths map[string][]string
	FrameBytes map[string][][]byte
}

type EpisodeStats ¶

type EpisodeStats map[string]FeatureStats

func ComputeEpisodeStats(in EpisodeInput, features map[string]FeatureDesc, opts Options) EpisodeStats

func SanitizeEpisodeStats(ep EpisodeStats) EpisodeStats

type FeatureDesc ¶

type FeatureDesc struct {
	DType string
	Shape []int
}

type FeatureStats ¶

type FeatureStats map[string]any

type ImageStat311 ¶

type ImageStat311 [][][]float64

ImageStat311 is the (3,1,1) nested layout used by LeRobot image/video stats.

func ImageStat311FromChannels(ch []float64) ImageStat311

func (s ImageStat311) Channels() []float64

type Mode ¶

type Mode int

Mode controls how image/video statistics are computed.

const (
	// ModeSampled matches official compute_episode_stats (estimate_num_samples heuristic).
	ModeSampled Mode = iota
	// ModeFull uses every frame path when computing image/video statistics.
	ModeFull
)

type Options ¶

type Options struct {
	Mode Mode
}