helper_utils

Classes

Constants
ExtractorConstants
RefinerConstants
UpSamplerConstants

Functions

timeit(func)
to_bf16_jax(data)
read_json([filename])	Reads a JSON file and returns the data as a dictionary.
write_json(data[, filename])	Writes a dictionary to a JSON file.
get_nv12(bgr_image)

Module Contents

class helper_utils.Constants

PROJECT_FOLDER = '/data/simaai/applications/RaftStereo'

DUMP = False

DEBUG = False

class helper_utils.ExtractorConstants

unpack_shapes = [7864320, 7864320, 3932160, 983040, 245760, 11796480, 2949120, 737280, 245760, 245760]

output_shapes = [[1, 80, 192, 256], [1, 80, 192, 256], [1, 80, 192, 128], [1, 40, 96, 128], [1, 20, 48, 128],...

ifm_size = 2949120

ofm_size = 36864000

class helper_utils.RefinerConstants

unpack_shapes = [3932160, 983040, 245760, 245760, 4423680, 245760]

ifm_size = 36864000

ofm_size = 4669440

iter_num = 12

class helper_utils.UpSamplerConstants

ifm_size = 4669440

ofm_size = 3932160

flowup_shape = [1, 80, 192, 8]

upmask_shape = [1, 80, 192, 144]

HEIGHT = 1280

WIDTH = 720

output_dim = 0

helper_utils.timeit(func)

helper_utils.to_bf16_jax(data)

helper_utils.read_json(filename='data.json')

Reads a JSON file and returns the data as a dictionary.

helper_utils.write_json(data, filename='data.json')

Writes a dictionary to a JSON file.

helper_utils.get_nv12(bgr_image)