SYMBOL INDEX (269 symbols across 29 files) FILE: lib/augmentation/event_augmentation.py function sample (line 8) | def sample(cdf, ts): function events_to_block (line 23) | def events_to_block(xs, ys, ts, ps): function merge_events (line 40) | def merge_events(event_sets): function add_random_events (line 60) | def add_random_events(xs, ys, ts, ps, to_add, sensor_resolution=None, function remove_events (line 94) | def remove_events(xs, ys, ts, ps, to_remove, add_noise=0): function add_correlated_events (line 118) | def add_correlated_events(xs, ys, ts, ps, to_add, sort=True, return_merg... function flip_events_x (line 159) | def flip_events_x(xs, ys, ts, ps, sensor_resolution=(180,240)): function flip_events_y (line 171) | def flip_events_y(xs, ys, ts, ps, sensor_resolution=(180,240)): function crop_events (line 183) | def crop_events(xs, ys, sensor_resolution, new_resolution): function rotate_events (line 195) | def rotate_events(xs, ys, sensor_resolution=(180,240), FILE: lib/contrast_max/events_cmax.py function get_hsv_shifted (line 14) | def get_hsv_shifted(): function grid_cmax (line 28) | def grid_cmax(xs, ys, ts, ps, roi_size=(20,20), step=None, warp=linvel_w... function segmentation_mask_from_d_iwe (line 78) | def segmentation_mask_from_d_iwe(d_iwe, th=None): function draw_objective_function (line 103) | def draw_objective_function(xs, ys, ts, ps, objective=variance_objective... function find_new_range (line 162) | def find_new_range(search_axes, param): function grid_search_optimisation (line 186) | def grid_search_optimisation(xs, ys, ts, ps, warp_function, objective_fu... function grid_search_initial (line 241) | def grid_search_initial(xs, ys, ts, ps, warp_function, objective_functio... function optimize_contrast (line 313) | def optimize_contrast(xs, ys, ts, ps, warp_function, objective, optimize... function optimize (line 348) | def optimize(xs, ys, ts, ps, warp, obj, numeric_grads=True, img_size=(18... function optimize_r2 (line 370) | def optimize_r2(xs, ys, ts, ps, warp, obj, numeric_grads=True, img_size=... FILE: lib/contrast_max/objectives.py class objective_function (line 10) | class objective_function(ABC): function __init__ (line 14) | def __init__(self, name="template", use_polarity=True, function evaluate_function (line 49) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps=N... function evaluate_gradient (line 78) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps=N... function iter_update (line 113) | def iter_update(self, params, pixel_crossings=None): function update_lifespan (line 129) | def update_lifespan(self, ts): function cut_events_to_lifespan (line 143) | def cut_events_to_lifespan(xs, ys, ts, ps, params, pixel_crossings, mini... function get_iwe (line 165) | def get_iwe(params, xs, ys, ts, ps, warpfunc, img_size, compute_gradient... class variance_objective (line 202) | class variance_objective(objective_function): method __init__ (line 206) | def __init__(self, adaptive_lifespan=False, minimum_events=10000): method evaluate_function (line 211) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 238) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class rms_objective (line 266) | class rms_objective(objective_function): method __init__ (line 270) | def __init__(self): method evaluate_function (line 276) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 291) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class sos_objective (line 308) | class sos_objective(objective_function): method __init__ (line 314) | def __init__(self, adaptive_lifespan=False, minimum_events=10000): method evaluate_function (line 325) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 339) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class soe_objective (line 358) | class soe_objective(objective_function): method __init__ (line 363) | def __init__(self): method evaluate_function (line 369) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 383) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class moa_objective (line 401) | class moa_objective(objective_function): method __init__ (line 406) | def __init__(self): method evaluate_function (line 412) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 425) | def evaluate_gradient(self, iwe=None, d_iwe=None, blur_sigma=None, sho... class isoa_objective (line 431) | class isoa_objective(objective_function): method __init__ (line 436) | def __init__(self, thresh=0.5): method evaluate_function (line 443) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 458) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class sosa_objective (line 478) | class sosa_objective(objective_function): method __init__ (line 483) | def __init__(self, p=3): method evaluate_function (line 490) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 505) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class zhu_timestamp_objective (line 524) | class zhu_timestamp_objective(objective_function): method __init__ (line 529) | def __init__(self): method evaluate_function (line 535) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 553) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... class r1_objective (line 560) | class r1_objective(objective_function): method __init__ (line 565) | def __init__(self, p=3): method evaluate_function (line 573) | def evaluate_function(self, params=None, xs=None, ys=None, ts=None, ps... method evaluate_gradient (line 591) | def evaluate_gradient(self, params=None, xs=None, ys=None, ts=None, ps... FILE: lib/contrast_max/warps.py class warp_function (line 6) | class warp_function(ABC): function __init__ (line 11) | def __init__(self, name, dims): function warp (line 22) | def warp(self, xs, ys, ts, ps, t0, params, compute_grad=False): class linvel_warp (line 44) | class linvel_warp(warp_function): method __init__ (line 48) | def __init__(self): method warp (line 51) | def warp(self, xs, ys, ts, ps, t0, params, compute_grad=False): class xyztheta_warp (line 63) | class xyztheta_warp(warp_function): method __init__ (line 68) | def __init__(self): method warp (line 71) | def warp(self, xs, ys, ts, ps, t0, params, compute_grad=False): class pure_rotation_warp (line 74) | class pure_rotation_warp(warp_function): method __init__ (line 79) | def __init__(self): function warp (line 82) | def warp(self, xs, ys, ts, ps, t0, params, compute_grad=False): FILE: lib/data_formats/add_hdf5_attribute.py function endswith (line 7) | def endswith(path, extensions): function get_filepaths_from_path_or_file (line 13) | def get_filepaths_from_path_or_file(path, extensions=[], datafile_extens... function add_attribute (line 28) | def add_attribute(h5_filepaths, group, attribute_name, attribute_value, ... FILE: lib/data_formats/data_providers.py class BaseDataLoader (line 5) | class BaseDataLoader(): method __init__ (line 6) | def __init__(self, data_root, iter_method='between_frames'): method __getitem__ (line 9) | def __getitem__(): method __len__ (line 12) | def __len__(self): FILE: lib/data_formats/data_utils.py function binary_search_h5_dset (line 4) | def binary_search_h5_dset(dset, x, l=None, r=None, side='left'): function binary_search_h5_timestamp (line 20) | def binary_search_h5_timestamp(hdf_path, l, r, x, side='left'): FILE: lib/data_formats/event_packagers.py class packager (line 6) | class packager(): method __init__ (line 13) | def __init__(self, name, output_path, max_buffer_size=1000000): method package_events (line 26) | def package_events(self, xs, ys, ts, ps): method package_image (line 38) | def package_image(self, frame, timestamp): method package_flow (line 48) | def package_flow(self, flow, timestamp): method add_metadata (line 58) | def add_metadata(self, num_events, num_pos, num_neg, method set_data_available (line 74) | def set_data_available(self, num_images, num_flow): class hdf5_packager (line 82) | class hdf5_packager(packager): method __init__ (line 86) | def __init__(self, output_path, max_buffer_size=1000000): method append_to_dataset (line 95) | def append_to_dataset(self, dataset, data): method package_events (line 101) | def package_events(self, xs, ys, ts, ps): method package_image (line 107) | def package_image(self, image, timestamp, img_idx): method package_flow (line 114) | def package_flow(self, flow_image, timestamp, flow_idx): method add_event_indices (line 120) | def add_event_indices(self): method add_metadata (line 138) | def add_metadata(self, num_pos, num_neg, method set_data_available (line 151) | def set_data_available(self, num_images, num_flow): FILE: lib/data_formats/h5_to_memmap.py class NpEncoder (line 7) | class NpEncoder(json.JSONEncoder): method default (line 8) | def default(self, obj): function find_safe_alternative (line 18) | def find_safe_alternative(output_base_path): function save_additional_data_as_mmap (line 27) | def save_additional_data_as_mmap(f, mmap_pth, data): function write_metadata (line 63) | def write_metadata(f, metadata_path): function h5_to_memmap (line 73) | def h5_to_memmap(h5_file_path, output_base_path, overwrite=True): FILE: lib/data_formats/read_events.py function compute_indices (line 5) | def compute_indices(event_stamps, frame_stamps): function read_memmap_events (line 19) | def read_memmap_events(memmap_path, skip_frames=1, return_events=False, ... function read_memmap_events_dict (line 95) | def read_memmap_events_dict(memmap_path, skip_frames=1, return_events=Fa... function read_h5_events (line 111) | def read_h5_events(hdf_path): function read_h5_event_components (line 125) | def read_h5_event_components(hdf_path): function read_h5_events_dict (line 138) | def read_h5_events_dict(hdf_path, read_frames=True): FILE: lib/data_formats/rosbag_to_h5.py function append_to_dataset (line 13) | def append_to_dataset(dataset, data): function timestamp_float (line 20) | def timestamp_float(ts): function get_rosbag_stats (line 24) | def get_rosbag_stats(bag, event_topic, image_topic=None, flow_topic=None): function extract_rosbag (line 43) | def extract_rosbag(rosbag_path, output_path, event_topic, image_topic=None, function extract_rosbags (line 142) | def extract_rosbags(rosbag_paths, output_dir, event_topic, image_topic, ... FILE: lib/data_loaders/base_dataset.py class BaseVoxelDataset (line 14) | class BaseVoxelDataset(Dataset): method get_frame (line 65) | def get_frame(self, index): method get_flow (line 72) | def get_flow(self, index): method get_events (line 79) | def get_events(self, idx0, idx1): method load_data (line 87) | def load_data(self, data_path): method find_ts_index (line 103) | def find_ts_index(self, timestamp): method ts (line 110) | def ts(self, index): method __init__ (line 117) | def __init__(self, data_path, transforms={}, sensor_resolution=None, n... method preprocess_events (line 210) | def preprocess_events(xs, ys, ts, ps): method __getitem__ (line 226) | def __getitem__(self, index, seed=None): method compute_between_frame_indices (line 322) | def compute_between_frame_indices(self): method compute_timeblock_indices (line 338) | def compute_timeblock_indices(self): method compute_k_indices (line 354) | def compute_k_indices(self): method compute_per_frame_indices (line 369) | def compute_per_frame_indices(self): method set_voxel_method (line 385) | def set_voxel_method(self, voxel_method): method __len__ (line 419) | def __len__(self): method get_event_indices (line 422) | def get_event_indices(self, index): method get_voxel_grid (line 433) | def get_voxel_grid(self, xs, ys, ts, ps, combined_voxel_channels=True): method transform_frame (line 457) | def transform_frame(self, frame, seed): method transform_voxel (line 471) | def transform_voxel(self, voxel, seed): method transform_flow (line 483) | def transform_flow(self, flow, seed): method size (line 497) | def size(self): method unpackage_events (line 505) | def unpackage_events(events): method collate_fn (line 513) | def collate_fn(data, event_keys=['events'], idx_keys=['events_batch_in... FILE: lib/data_loaders/data_augmentation.py class Compose (line 6) | class Compose(object): method __init__ (line 16) | def __init__(self, transforms): method __call__ (line 22) | def __call__(self, x, is_flow=False): method __repr__ (line 33) | def __repr__(self): class CenterCrop (line 42) | class CenterCrop(object): method __init__ (line 47) | def __init__(self, size, preserve_mosaicing_pattern=False): method __call__ (line 55) | def __call__(self, x, is_flow=False): method __repr__ (line 78) | def __repr__(self): class RobustNorm (line 82) | class RobustNorm(object): method __init__ (line 89) | def __init__(self, low_perc=0, top_perc=95): method percentile (line 94) | def percentile(t, q): method __call__ (line 115) | def __call__(self, x, is_flow=False): method __repr__ (line 132) | def __repr__(self): FILE: lib/data_loaders/data_util.py function concatenate_subfolders (line 11) | def concatenate_subfolders(data_file, dataset, dataset_kwargs): function concatenate_datasets (line 29) | def concatenate_datasets(data_file, dataset_type, dataset_kwargs=None): function concatenate_memmap_datasets (line 50) | def concatenate_memmap_datasets(data_file, dataset_type, dataset_kwargs): FILE: lib/data_loaders/dataloader_util.py function unpack_batched_events (line 3) | def unpack_batched_events(events, batch_indices): FILE: lib/data_loaders/hdf5_dataset.py class DynamicH5Dataset (line 6) | class DynamicH5Dataset(BaseVoxelDataset): method get_frame (line 12) | def get_frame(self, index): method get_flow (line 15) | def get_flow(self, index): method get_events (line 18) | def get_events(self, idx0, idx1): method load_data (line 25) | def load_data(self, data_path): method find_ts_index (line 53) | def find_ts_index(self, timestamp): method ts (line 57) | def ts(self, index): method compute_frame_indices (line 60) | def compute_frame_indices(self): FILE: lib/data_loaders/memmap_dataset.py class MemMapDataset (line 5) | class MemMapDataset(BaseVoxelDataset): method get_frame (line 11) | def get_frame(self, index): method get_flow (line 15) | def get_flow(self, index): method get_events (line 19) | def get_events(self, idx0, idx1): method load_data (line 27) | def load_data(self, data_path, timestamp_fname="timestamps.npy", image... method find_ts_index (line 81) | def find_ts_index(self, timestamp): method ts (line 85) | def ts(self, index): method infer_resolution (line 88) | def infer_resolution(self): method find_config (line 96) | def find_config(self, data_path): FILE: lib/data_loaders/npy_dataset.py class NpyDataset (line 4) | class NpyDataset(BaseVoxelDataset): method get_frame (line 10) | def get_frame(self, index): method get_flow (line 13) | def get_flow(self, index): method get_events (line 16) | def get_events(self, idx0, idx1): method load_data (line 23) | def load_data(self, data_path): method find_ts_index (line 45) | def find_ts_index(self, timestamp): method ts (line 49) | def ts(self, index): method compute_frame_indices (line 52) | def compute_frame_indices(self): FILE: lib/representations/image.py function events_to_image (line 5) | def events_to_image(xs, ys, ps, sensor_size=(180, 240), interpolation=No... function events_to_image_torch (line 46) | def events_to_image_torch(xs, ys, ps, function interpolate_to_image (line 102) | def interpolate_to_image(pxs, pys, dxs, dys, weights, img): function interpolate_to_derivative_img (line 117) | def interpolate_to_derivative_img(pxs, pys, dxs, dys, d_img, w1, w2): function image_to_event_weights (line 138) | def image_to_event_weights(xs, ys, img): function events_to_image_drv (line 162) | def events_to_image_drv(xn, yn, pn, jacobian_xn, jacobian_yn, function events_to_timestamp_image (line 219) | def events_to_timestamp_image(xn, yn, ts, pn, function events_to_timestamp_image_torch (line 286) | def events_to_timestamp_image_torch(xs, ys, ts, ps, class TimestampImage (line 355) | class TimestampImage: method __init__ (line 357) | def __init__(self, sensor_size): method set_init (line 362) | def set_init(self, value): method add_event (line 365) | def add_event(self, x, y, t, p): method add_events (line 368) | def add_events(self, xs, ys, ts, ps): method get_image (line 372) | def get_image(self): class EventImage (line 379) | class EventImage: method __init__ (line 381) | def __init__(self, sensor_size): method add_event (line 386) | def add_event(self, x, y, t, p): method add_events (line 389) | def add_events(self, xs, ys, ts, ps): method get_image (line 393) | def get_image(self): FILE: lib/representations/voxel_grid.py function get_voxel_grid_as_image (line 9) | def get_voxel_grid_as_image(voxelgrid): function plot_voxel_grid (line 26) | def plot_voxel_grid(voxelgrid, cmap='gray'): function voxel_grids_fixed_n_torch (line 37) | def voxel_grids_fixed_n_torch(xs, ys, ts, ps, B, n, sensor_size=(180, 24... function voxel_grids_fixed_t_torch (line 59) | def voxel_grids_fixed_t_torch(xs, ys, ts, ps, B, t, sensor_size=(180, 24... function events_to_voxel_timesync_torch (line 82) | def events_to_voxel_timesync_torch(xs, ys, ts, ps, B, t0, t1, device=Non... function events_to_voxel_torch (line 114) | def events_to_voxel_torch(xs, ys, ts, ps, B, device=None, sensor_size=(1... function events_to_neg_pos_voxel_torch (line 155) | def events_to_neg_pos_voxel_torch(xs, ys, ts, ps, B, device=None, function events_to_voxel (line 184) | def events_to_voxel(xs, ys, ts, ps, B, sensor_size=(180, 240), temporal_... function events_to_neg_pos_voxel (line 219) | def events_to_neg_pos_voxel(xs, ys, ts, ps, B, FILE: lib/transforms/optic_flow.py function warp_events_flow_torch (line 5) | def warp_events_flow_torch(xt, yt, tt, pt, flow_field, t0=None, FILE: lib/util/event_util.py function infer_resolution (line 5) | def infer_resolution(xs, ys): function events_bounds_mask (line 15) | def events_bounds_mask(xs, ys, x_min, x_max, y_min, y_max): function cut_events_to_lifespan (line 30) | def cut_events_to_lifespan(xs, ys, ts, ps, params, function clip_events_to_bounds (line 61) | def clip_events_to_bounds(xs, ys, ts, ps, bounds, set_zero=False): function get_events_from_mask (line 96) | def get_events_from_mask(mask, xs, ys): function binary_search_h5_dset (line 111) | def binary_search_h5_dset(dset, x, l=None, r=None, side='left'): function binary_search_h5_timestamp (line 137) | def binary_search_h5_timestamp(hdf_path, l, r, x, side='left'): function binary_search_torch_tensor (line 141) | def binary_search_torch_tensor(t, l, r, x, side='left'): function remove_hot_pixels (line 166) | def remove_hot_pixels(xs, ys, ts, ps, sensor_size=(180, 240), num_hot=50): FILE: lib/util/util.py function ensure_dir (line 15) | def ensure_dir(dirname): function read_json (line 26) | def read_json(fname): function write_json (line 32) | def write_json(content, fname): function inf_loop (line 38) | def inf_loop(data_loader): function optimal_crop_size (line 44) | def optimal_crop_size(max_size, max_subsample_factor, safety_margin=0): class CropParameters (line 54) | class CropParameters: method __init__ (line 61) | def __init__(self, width, height, num_encoders, safety_margin=0): method crop (line 83) | def crop(self, img): function format_power (line 87) | def format_power(size): function plot_image (line 96) | def plot_image(image, lognorm=False, cmap='gray', bbox=None, ticks=False... function plot_image_grid (line 127) | def plot_image_grid(images, grid_shape=None, lognorm=False, function save_image (line 164) | def save_image(image, fname=None, lognorm=False, cmap='gray', bbox=None,... function flow2bgr_np (line 182) | def flow2bgr_np(disp_x, disp_y, max_magnitude=None): FILE: lib/visualization/draw_event_stream.py function plot_events_sliding (line 15) | def plot_events_sliding(xs, ys, ts, ps, args, frames=[], frame_ts=[]): function plot_voxel_grid (line 75) | def plot_voxel_grid(xs, ys, ts, ps, bins=5, frames=[], frame_ts=[], function plot_events (line 152) | def plot_events(xs, ys, ts, ps, save_path=None, num_compress='auto', num... function plot_between_frames (line 278) | def plot_between_frames(xs, ys, ts, ps, frames, frame_event_idx, args, p... FILE: lib/visualization/draw_event_stream_mayavi.py function plot_events_sliding (line 17) | def plot_events_sliding(xs, ys, ts, ps, args, dt=None, sdt=None, frames=... function plot_voxel_grid (line 99) | def plot_voxel_grid(xs, ys, ts, ps, bins=5, frames=[], frame_ts=[], function plot_events (line 161) | def plot_events(xs, ys, ts, ps, save_path=None, num_compress='auto', num... function plot_between_frames (line 234) | def plot_between_frames(xs, ys, ts, ps, frames, frame_event_idx, args, p... FILE: lib/visualization/draw_flow.py function motion_compensate (line 15) | def motion_compensate(xs, ys, ts, ps, flow, fname="/tmp/img.png", crop=N... function plot_flow_and_events (line 28) | def plot_flow_and_events(xs, ys, ts, ps, flow, save_path=None, function plot_between_frames (line 128) | def plot_between_frames(xs, ys, ts, ps, flows, flow_imgs, flow_ts, args,... FILE: lib/visualization/visualization_utils.py function frame_stamps_to_start_end (line 4) | def frame_stamps_to_start_end(stamps): function get_frame_indices (line 10) | def get_frame_indices(ts, frame_ts): function crop_to_size (line 14) | def crop_to_size(crop): function parse_crop (line 17) | def parse_crop(cropstr): function ensure_dir (line 34) | def ensure_dir(file_path): FILE: lib/visualization/visualizers.py class Visualizer (line 16) | class Visualizer(): method __init__ (line 18) | def __init__(self): method plot_events (line 21) | def plot_events(self, data, save_path, **kwargs): method unpackage_events (line 25) | def unpackage_events(events): class TimeStampImageVisualizer (line 28) | class TimeStampImageVisualizer(Visualizer): method __init__ (line 30) | def __init__(self, sensor_size): method plot_events (line 34) | def plot_events(self, data, save_path, **kwargs): class EventImageVisualizer (line 45) | class EventImageVisualizer(Visualizer): method __init__ (line 47) | def __init__(self, sensor_size): method plot_events (line 50) | def plot_events(self, data, save_path, **kwargs): class EventsVisualizer (line 63) | class EventsVisualizer(Visualizer): method __init__ (line 65) | def __init__(self, sensor_size): method plot_events (line 68) | def plot_events(self, data, save_path, class VoxelVisualizer (line 206) | class VoxelVisualizer(Visualizer): method __init__ (line 208) | def __init__(self, sensor_size): method increase_brightness (line 212) | def increase_brightness(rgb, increase=0.5): method plot_events (line 222) | def plot_events(self, data, save_path, bins=5, crop=None, elev=0, azim... FILE: visualize_voxel.py function plot_events_sliding (line 7) | def plot_events_sliding(xs, ys, ts, ps, args, frames=None, frame_ts=None):