SYMBOL INDEX (342 symbols across 33 files)

FILE: dvs/dataset.py
  function get_data_loader (line 26) | def get_data_loader(cf, no_flo = False):
  function get_dataset (line 34) | def get_dataset(cf, no_flo = False):
  function get_inference_data_loader (line 52) | def get_inference_data_loader(cf, data_path, no_flo = False):
  function get_inference_dataset (line 57) | def get_inference_dataset(cf, data_path, no_flo = False):
  function _data_transforms (line 66) | def _data_transforms():
  class DVS_data (line 77) | class DVS_data():
    method __init__ (line 78) | def __init__(self):
  class Dataset_Gyro (line 87) | class Dataset_Gyro(Dataset):
    method __init__ (line 88) | def __init__(self, path, sample_freq = 33*1000000, number_real = 10, t...
    method process_one_video (line 122) | def process_one_video(self, path):
    method generate_quaternions (line 154) | def generate_quaternions(self, dvs_data):
    method load_flo (line 179) | def load_flo(self, idx, first_id):
    method load_real_projections (line 195) | def load_real_projections(self, idx, first_id):
    method __getitem__ (line 203) | def __getitem__(self, idx):
    method __len__ (line 212) | def __len__(self):
    method get_virtual_data (line 215) | def get_virtual_data(self, virtual_queue, real_queue_idx, pre_times, c...
    method update_virtual_queue (line 232) | def update_virtual_queue(self, batch_size, virtual_queue, out, times):
    method random_init_virtual_queue (line 244) | def random_init_virtual_queue(self, batch_size, real_postion, times):
    method get_data_at_timestamp (line 259) | def get_data_at_timestamp(self, gyro_data, ois_data, time_stamp, quat_...
    method get_ois_at_timestamp (line 264) | def get_ois_at_timestamp(self, ois_data, time_stamp):
  function get_timestamp (line 269) | def get_timestamp(frame_data, idx):
  function preprocess_gyro (line 275) | def preprocess_gyro(gyro, extend = 200):
  function LoadFlow (line 286) | def LoadFlow(path):
  function get_virtual_at_timestamp (line 293) | def get_virtual_at_timestamp(virtual_queue, real_queue, time_stamp, time...

FILE: dvs/flownet2/datasets.py
  class StaticRandomCrop (line 13) | class StaticRandomCrop(object):
    method __init__ (line 14) | def __init__(self, image_size, crop_size):
    method __call__ (line 20) | def __call__(self, img):
  class StaticCenterCrop (line 23) | class StaticCenterCrop(object):
    method __init__ (line 24) | def __init__(self, image_size, crop_size):
    method __call__ (line 27) | def __call__(self, img):
  class Padding (line 30) | class Padding(object):
    method __init__ (line 31) | def __init__(self, image_size, pad_size):
    method __call__ (line 34) | def __call__(self, img):
  class MpiSintel (line 39) | class MpiSintel(data.Dataset):
    method __init__ (line 40) | def __init__(self, args, is_cropped = False, root = '', dstype = 'clea...
    method __getitem__ (line 85) | def __getitem__(self, index):
    method __len__ (line 112) | def __len__(self):
  class MpiSintelClean (line 115) | class MpiSintelClean(MpiSintel):
    method __init__ (line 116) | def __init__(self, args, is_cropped = False, root = '', replicates = 1):
  class MpiSintelFinal (line 119) | class MpiSintelFinal(MpiSintel):
    method __init__ (line 120) | def __init__(self, args, is_cropped = False, root = '', replicates = 1):
  class FlyingChairs (line 123) | class FlyingChairs(data.Dataset):
    method __init__ (line 124) | def __init__(self, args, is_cropped, root = '/path/to/FlyingChairs_rel...
    method __getitem__ (line 155) | def __getitem__(self, index):
    method __len__ (line 181) | def __len__(self):
  class FlyingThings (line 184) | class FlyingThings(data.Dataset):
    method __init__ (line 185) | def __init__(self, args, is_cropped, root = '/path/to/flyingthings3d',...
    method __getitem__ (line 222) | def __getitem__(self, index):
    method __len__ (line 248) | def __len__(self):
  class FlyingThingsClean (line 251) | class FlyingThingsClean(FlyingThings):
    method __init__ (line 252) | def __init__(self, args, is_cropped = False, root = '', replicates = 1):
  class FlyingThingsFinal (line 255) | class FlyingThingsFinal(FlyingThings):
    method __init__ (line 256) | def __init__(self, args, is_cropped = False, root = '', replicates = 1):
  class ChairsSDHom (line 259) | class ChairsSDHom(data.Dataset):
    method __init__ (line 260) | def __init__(self, args, is_cropped, root = '/path/to/chairssdhom/data...
    method __getitem__ (line 291) | def __getitem__(self, index):
    method __len__ (line 318) | def __len__(self):
  class ChairsSDHomTrain (line 321) | class ChairsSDHomTrain(ChairsSDHom):
    method __init__ (line 322) | def __init__(self, args, is_cropped = False, root = '', replicates = 1):
  class ChairsSDHomTest (line 325) | class ChairsSDHomTest(ChairsSDHom):
    method __init__ (line 326) | def __init__(self, args, is_cropped = False, root = '', replicates = 1):
  class ImagesFromFolder (line 329) | class ImagesFromFolder(data.Dataset):
    method __init__ (line 330) | def __init__(self, args, is_cropped, root = '/path/to/frames/only/fold...
    method __getitem__ (line 354) | def __getitem__(self, index):
    method __len__ (line 373) | def __len__(self):
  class Google (line 377) | class Google(data.Dataset):
    method __init__ (line 378) | def __init__(self, args, is_cropped = False, root = '', dstype = 'fram...
    method __getitem__ (line 412) | def __getitem__(self, index):
    method __len__ (line 434) | def __len__(self):

FILE: dvs/flownet2/losses.py
  function EPE (line 11) | def EPE(input_flow, target_flow):
  class L1 (line 14) | class L1(nn.Module):
    method __init__ (line 15) | def __init__(self):
    method forward (line 17) | def forward(self, output, target):
  class L2 (line 21) | class L2(nn.Module):
    method __init__ (line 22) | def __init__(self):
    method forward (line 24) | def forward(self, output, target):
  class L1Loss (line 28) | class L1Loss(nn.Module):
    method __init__ (line 29) | def __init__(self, args):
    method forward (line 35) | def forward(self, output, target):
  class L2Loss (line 40) | class L2Loss(nn.Module):
    method __init__ (line 41) | def __init__(self, args):
    method forward (line 47) | def forward(self, output, target):
  class MultiScale (line 52) | class MultiScale(nn.Module):
    method __init__ (line 53) | def __init__(self, args, startScale = 4, numScales = 5, l_weight= 0.32...
    method forward (line 72) | def forward(self, output, target):

FILE: dvs/flownet2/main.py
  function inference (line 22) | def inference(args, epoch, data_path, data_loader, model, offset=0):
  class Model (line 183) | class Model(nn.Module):
    method __init__ (line 184) | def __init__(self, args):
    method forward (line 189) | def forward(self, data):

FILE: dvs/flownet2/models.py
  class FlowNet2 (line 30) | class FlowNet2(nn.Module):
    method __init__ (line 32) | def __init__(self, args, batchNorm=False, div_flow = 20.):
    method init_deconv_bilinear (line 104) | def init_deconv_bilinear(self, weight):
    method forward (line 120) | def forward(self, inputs):
  class FlowNet2C (line 187) | class FlowNet2C(FlowNetC.FlowNetC):
    method __init__ (line 188) | def __init__(self, args, batchNorm=False, div_flow=20):
    method forward (line 192) | def forward(self, inputs):
  class FlowNet2S (line 255) | class FlowNet2S(FlowNetS.FlowNetS):
    method __init__ (line 256) | def __init__(self, args, batchNorm=False, div_flow=20):
    method forward (line 261) | def forward(self, inputs):
  class FlowNet2SD (line 301) | class FlowNet2SD(FlowNetSD.FlowNetSD):
    method __init__ (line 302) | def __init__(self, args, batchNorm=False, div_flow=20):
    method forward (line 307) | def forward(self, inputs):
  class FlowNet2CS (line 353) | class FlowNet2CS(nn.Module):
    method __init__ (line 355) | def __init__(self, args, batchNorm=False, div_flow = 20.):
    method forward (line 392) | def forward(self, inputs):
  class FlowNet2CSS (line 418) | class FlowNet2CSS(nn.Module):
    method __init__ (line 420) | def __init__(self, args, batchNorm=False, div_flow = 20.):
    method forward (line 469) | def forward(self, inputs):

FILE: dvs/flownet2/networks/FlowNetC.py
  class FlowNetC (line 13) | class FlowNetC(nn.Module):
    method __init__ (line 14) | def __init__(self,args, batchNorm=True, div_flow = 20):
    method forward (line 71) | def forward(self, x):

FILE: dvs/flownet2/networks/FlowNetFusion.py
  class FlowNetFusion (line 11) | class FlowNetFusion(nn.Module):
    method __init__ (line 12) | def __init__(self,args, batchNorm=True):
    method forward (line 47) | def forward(self, x):

FILE: dvs/flownet2/networks/FlowNetS.py
  class FlowNetS (line 15) | class FlowNetS(nn.Module):
    method __init__ (line 16) | def __init__(self, args, input_channels = 12, batchNorm=True):
    method forward (line 60) | def forward(self, x):

FILE: dvs/flownet2/networks/FlowNetSD.py
  class FlowNetSD (line 11) | class FlowNetSD(nn.Module):
    method __init__ (line 12) | def __init__(self, args, batchNorm=True):
    method forward (line 66) | def forward(self, x):

FILE: dvs/flownet2/networks/channelnorm_package/channelnorm.py
  class ChannelNormFunction (line 5) | class ChannelNormFunction(Function):
    method forward (line 8) | def forward(ctx, input1, norm_deg=2):
    method backward (line 20) | def backward(ctx, grad_output):
  class ChannelNorm (line 31) | class ChannelNorm(Module):
    method __init__ (line 33) | def __init__(self, norm_deg=2):
    method forward (line 37) | def forward(self, input1):

FILE: dvs/flownet2/networks/channelnorm_package/channelnorm_cuda.cc
  function channelnorm_cuda_forward (line 6) | int channelnorm_cuda_forward(
  function channelnorm_cuda_backward (line 16) | int channelnorm_cuda_backward(
  function PYBIND11_MODULE (line 27) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: dvs/flownet2/networks/correlation_package/correlation.py
  class CorrelationFunction (line 6) | class CorrelationFunction(Function):
    method forward (line 9) | def forward(ctx, input1, input2, pad_size=3, kernel_size=3, max_displa...
    method backward (line 30) | def backward(ctx, grad_output):
  class Correlation (line 46) | class Correlation(Module):
    method __init__ (line 47) | def __init__(self, pad_size=0, kernel_size=0, max_displacement=0, stri...
    method forward (line 56) | def forward(self, input1, input2):

FILE: dvs/flownet2/networks/correlation_package/correlation_cuda.cc
  function correlation_forward_cuda (line 10) | int correlation_forward_cuda(at::Tensor& input1, at::Tensor& input2, at:...
  function correlation_backward_cuda (line 89) | int correlation_backward_cuda(at::Tensor& input1, at::Tensor& input2, at...
  function PYBIND11_MODULE (line 169) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: dvs/flownet2/networks/resample2d_package/resample2d.py
  class Resample2dFunction (line 5) | class Resample2dFunction(Function):
    method forward (line 8) | def forward(ctx, input1, input2, kernel_size=1, bilinear= True):
    method backward (line 25) | def backward(ctx, grad_output):
  class Resample2d (line 40) | class Resample2d(Module):
    method __init__ (line 42) | def __init__(self, kernel_size=1, bilinear = True):
    method forward (line 47) | def forward(self, input1, input2):

FILE: dvs/flownet2/networks/resample2d_package/resample2d_cuda.cc
  function resample2d_cuda_forward (line 6) | int resample2d_cuda_forward(
  function resample2d_cuda_backward (line 15) | int resample2d_cuda_backward(
  function PYBIND11_MODULE (line 28) | PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {

FILE: dvs/flownet2/networks/submodules.py
  function conv (line 7) | def conv(batchNorm, in_planes, out_planes, kernel_size=3, stride=1):
  function i_conv (line 20) | def i_conv(batchNorm, in_planes, out_planes, kernel_size=3, stride=1, bi...
  function predict_flow (line 31) | def predict_flow(in_planes):
  function deconv (line 34) | def deconv(in_planes, out_planes):
  class tofp16 (line 40) | class tofp16(nn.Module):
    method __init__ (line 41) | def __init__(self):
    method forward (line 44) | def forward(self, input):
  class tofp32 (line 48) | class tofp32(nn.Module):
    method __init__ (line 49) | def __init__(self):
    method forward (line 52) | def forward(self, input):
  function init_deconv_bilinear (line 56) | def init_deconv_bilinear(weight):
  function save_grad (line 72) | def save_grad(grads, name):

FILE: dvs/flownet2/utils/flow_utils.py
  function readFlow (line 7) | def readFlow(fn):
  function writeFlow (line 28) | def writeFlow(filename,uv,v=None):
  function visulize_flow_file (line 62) | def visulize_flow_file(flow_filename, save_dir=None):
  function flow2img (line 72) | def flow2img(flow_data):
  function compute_color (line 112) | def compute_color(u, v):
  function make_color_wheel (line 157) | def make_color_wheel():

FILE: dvs/flownet2/utils/frame_utils.py
  function read_gen (line 6) | def read_gen(file_name):

FILE: dvs/flownet2/utils/param_utils.py
  function parse_flownetc (line 5) | def parse_flownetc(modules, weights, biases):
  function parse_flownets (line 51) | def parse_flownets(modules, weights, biases, param_prefix='net2_'):
  function parse_flownetsonly (line 104) | def parse_flownetsonly(modules, weights, biases, param_prefix=''):
  function parse_flownetsd (line 156) | def parse_flownetsd(modules, weights, biases, param_prefix='netsd_'):
  function parse_flownetfusion (line 214) | def parse_flownetfusion(modules, weights, biases, param_prefix='fuse_'):

FILE: dvs/flownet2/utils/tools.py
  function datestr (line 13) | def datestr():
  function module_to_dict (line 18) | def module_to_dict(module, exclude=[]):
  class TimerBlock (line 24) | class TimerBlock:
    method __init__ (line 25) | def __init__(self, title):
    method __enter__ (line 28) | def __enter__(self):
    method __exit__ (line 32) | def __exit__(self, exc_type, exc_value, traceback):
    method log (line 42) | def log(self, string):
    method log2file (line 50) | def log2file(self, fid, string):
  function add_arguments_for_module (line 55) | def add_arguments_for_module(parser, module, argument_for_class, default...
  function kwargs_from_args (line 84) | def kwargs_from_args(args, argument_for_class):
  function format_dictionary_of_losses (line 88) | def format_dictionary_of_losses(labels, values):
  class IteratorTimer (line 98) | class IteratorTimer():
    method __init__ (line 99) | def __init__(self, iterable):
    method __iter__ (line 103) | def __iter__(self):
    method __len__ (line 106) | def __len__(self):
    method __next__ (line 109) | def __next__(self):
  function gpumemusage (line 117) | def gpumemusage():
  function update_hyperparameter_schedule (line 131) | def update_hyperparameter_schedule(args, epoch, global_iteration, optimi...
  function save_checkpoint (line 138) | def save_checkpoint(state, is_best, path, prefix, filename='checkpoint.p...

FILE: dvs/gyro/gyro_function.py
  function get_static (line 7) | def get_static(height = 1080, width = 1920, ratio = 0.1):
  function norm_quat (line 24) | def norm_quat(quat):
  function torch_norm_quat (line 34) | def torch_norm_quat(quat, USE_CUDA = True):
  function ConvertAxisAngleToQuaternion (line 53) | def ConvertAxisAngleToQuaternion(axis, angle):
  function ConvertAxisAngleToQuaternion_no_angle (line 62) | def ConvertAxisAngleToQuaternion_no_angle(axis):
  function torch_ConvertAxisAngleToQuaternion (line 72) | def torch_ConvertAxisAngleToQuaternion(axis, USE_CUDA = True):
  function ConvertQuaternionToAxisAngle (line 90) | def ConvertQuaternionToAxisAngle(quat):
  function ConvertQuaternionToAxisAngle_no_angle (line 104) | def ConvertQuaternionToAxisAngle_no_angle(quat):
  function torch_ConvertQuaternionToAxisAngle (line 115) | def torch_ConvertQuaternionToAxisAngle(quat, USE_CUDA = True):
  function train_ConvertQuaternionToAxisAngle (line 129) | def train_ConvertQuaternionToAxisAngle(quat):
  function AngularVelocityToQuat (line 134) | def AngularVelocityToQuat(angular_v, dt):
  function QuaternionProduct (line 144) | def QuaternionProduct(q1, q2):
  function torch_QuaternionProduct (line 163) | def torch_QuaternionProduct(q1, q2, USE_CUDA = True):
  function ProcessGyroRotation (line 188) | def ProcessGyroRotation(gyro_data):
  function QuaternionReciprocal (line 199) | def QuaternionReciprocal(q):
  function torch_QuaternionReciprocal (line 203) | def torch_QuaternionReciprocal(q,  USE_CUDA = True):
  function ProcessGyroData (line 210) | def ProcessGyroData(gyro_data):
  function SlerpWithDefault (line 221) | def SlerpWithDefault(q1, q2, t, q_default):
  function GetGyroAtTimeStamp (line 260) | def GetGyroAtTimeStamp(gyro_data, timestamp):
  function train_GetGyroAtTimeStamp (line 275) | def train_GetGyroAtTimeStamp(gyro_data, timestamp, check = False):
  function FindOISAtTimeStamp (line 291) | def FindOISAtTimeStamp(ois_log, time):
  function GetMetadata (line 311) | def GetMetadata(frame_data, frame_index, result_poses = {} ):
  function GetProjections (line 330) | def GetProjections(static_options, metadata, quats_data, ois_data,  no_s...
  function GetRealProjection (line 344) | def GetRealProjection(static_options, quats_data, ois_data, fov, timesta...
  function GetProjectionHomography (line 354) | def GetProjectionHomography(rot, fov, offset, width, height):
  function torch_GetProjectionHomography (line 365) | def torch_GetProjectionHomography(rot, fov, width, height, USE_CUDA = Tr...
  function ConvertQuaternionToRotationMatrix (line 381) | def ConvertQuaternionToRotationMatrix(quat):
  function torch_ConvertQuaternionToRotationMatrix (line 399) | def torch_ConvertQuaternionToRotationMatrix(quat, USE_CUDA = True):
  function ConvertRotationMatrixToQuaternion (line 422) | def ConvertRotationMatrixToQuaternion(m):
  function GetIntrinsics (line 450) | def GetIntrinsics(focal_length, offset, width, height):
  function GetVirtualProjection (line 459) | def GetVirtualProjection(static_options, result_pose, metadata, frame_in...
  function torch_GetVirtualProjection (line 470) | def torch_GetVirtualProjection(static_options, quat, virtual_fov = 1.27):
  function GetForwardGrid (line 476) | def GetForwardGrid(static_options, real_projections, virtual_projection):
  function torch_GetForwardGrid (line 498) | def torch_GetForwardGrid(static_options, real_projections, virtual_proje...
  function GetWarpingFlow (line 530) | def GetWarpingFlow(real_projections_src, real_projections_dst, num_rows,...
  function torch_GetWarpingFlow (line 549) | def torch_GetWarpingFlow(static_options, real_projections_src, real_proj...
  function GetHomographyTransformFromProjections (line 581) | def GetHomographyTransformFromProjections(proj_src, proj_dst):
  function torch_GetHomographyTransformFromProjections (line 584) | def torch_GetHomographyTransformFromProjections(proj_src, proj_dst):
  function ApplyTransform (line 587) | def ApplyTransform(transform, point):
  function torch_ApplyTransform (line 594) | def torch_ApplyTransform(transform, point):
  function CenterZoom (line 601) | def CenterZoom(grid, ratio):

FILE: dvs/gyro/gyro_io.py
  function load_gyro_mesh (line 13) | def load_gyro_mesh(input_name):
  function get_grid (line 19) | def get_grid(static_options, frame_data, quats_data, ois_data, virtual_d...
  function get_rotations (line 34) | def get_rotations(frame_data, quats_data, ois_data, num_frames):
  function visual_rotation (line 50) | def visual_rotation(rotations_real, lens_offsets_real, rotations_virtual...
  function LoadOISData (line 106) | def LoadOISData(ois_name):
  function LoadFrameData (line 111) | def LoadFrameData(frame_log_name):
  function LoadGyroData (line 117) | def LoadGyroData(gyro_log_name):
  function LoadStabResult (line 126) | def LoadStabResult(input_name):
  function ReadLine (line 142) | def ReadLine(fid):
  function str2num (line 162) | def str2num(string):

FILE: dvs/inference.py
  function run (line 29) | def run(model, loader, cf, USE_CUDA=True):
  function inference (line 120) | def inference(cf, data_path, USE_CUDA):
  function visual_result (line 173) | def visual_result(cf, data, video_name, virtual_queue, virtual_queue2 = ...
  function main (line 187) | def main(args = None):

FILE: dvs/load_frame_sensor_data.py
  function run (line 31) | def run(loader, cf, USE_CUDA=True):
  function inference (line 68) | def inference(cf, data_path, USE_CUDA):
  function main (line 97) | def main(args = None):

FILE: dvs/loss.py
  class C2_Smooth_loss (line 21) | class C2_Smooth_loss(torch.nn.Module):
    method __init__ (line 22) | def __init__(self):
    method forward (line 26) | def forward(self, Qt, Qt_1, Qt_2):
  class C1_Smooth_loss (line 30) | class C1_Smooth_loss(torch.nn.Module):
    method __init__ (line 31) | def __init__(self):
    method forward (line 35) | def forward(self, v_r_axis, v_axis_t_1 = None, real_postion = None):
  class Follow_loss (line 40) | class Follow_loss(torch.nn.Module):
    method __init__ (line 41) | def __init__(self):
    method forward (line 45) | def forward(self, virtual_quat, real_quat, real_postion = None):
  class Stay_loss (line 50) | class Stay_loss(torch.nn.Module):
    method __init__ (line 51) | def __init__(self):
    method forward (line 55) | def forward(self, virtual_quat):
  class Angle_loss (line 59) | class Angle_loss(torch.nn.Module):
    method __init__ (line 60) | def __init__(self):
    method forward (line 64) | def forward(self, Q1, Q2, threshold = 0.5236, logistic_beta1 = 100):
  class Optical_loss (line 73) | class Optical_loss(torch.nn.Module):
    method __init__ (line 74) | def __init__(self):
    method forward (line 79) | def forward(self, Vt, Vt_1, flo, flo_back, real_projection_t, real_pro...
  function get_mesh (line 122) | def get_mesh(height = 270, width = 480, USE_CUDA = True):
  class Undefine_loss (line 132) | class Undefine_loss(torch.nn.Module):
    method __init__ (line 133) | def __init__(self, ratio = 0.08, inner_ratio = 0.04, USE_CUDA = True):
    method forward (line 153) | def forward(self, Vt, Rt, ratio = 0.04):
    method get_loss (line 177) | def get_loss(self, p):

FILE: dvs/metrics.py
  function _pickle_keypoints (line 19) | def _pickle_keypoints(point):
  function crop_metric (line 30) | def crop_metric(M):
  function get_scale (line 39) | def get_scale(M):
  function get_rescale_matrix (line 53) | def get_rescale_matrix(M, sx, sy):
  function metrics (line 64) | def metrics(in_src, out_src, package, crop_scale = False, re_compute = F...
  function crop_rm_outlier (line 294) | def crop_rm_outlier(crop):

FILE: dvs/model.py
  class LayerLSTM (line 16) | class LayerLSTM(nn.Module):
    method __init__ (line 17) | def __init__(self, input_size, hidden_size, bias):
    method init_hidden (line 22) | def init_hidden(self, batch_size):
    method forward (line 26) | def forward(self, x):
  class LayerCNN (line 31) | class LayerCNN(nn.Module):
    method __init__ (line 32) | def __init__(self, in_channel, out_channel, kernel_size, stride, paddi...
    method forward (line 43) | def forward(self, x):
  class LayerFC (line 52) | class LayerFC(nn.Module):
    method __init__ (line 53) | def __init__(self, in_features, out_features, bias, drop_out=0, activa...
    method forward (line 61) | def forward(self, x):
  class Net (line 71) | class Net(nn.Module):
    method __init__ (line 72) | def __init__(self, cf):
    method init_hidden (line 160) | def init_hidden(self, batch_size):
    method forward (line 164) | def forward(self, x, flo, ois):
  class Model (line 178) | class Model():
    method __init__ (line 179) | def __init__(self, cf):
    method loss (line 203) | def loss(
    method init_weights (line 243) | def init_weights(self, cf):
    method save_checkpoint (line 258) | def save_checkpoint(self, epoch = 0, optimizer=None):
  class UNet (line 272) | class UNet(nn.Module):
    method __init__ (line 273) | def __init__(self, n_channels = 4, n_classes = 16, bilinear=True):
    method forward (line 288) | def forward(self, x, x_back = None):
  class DoubleConv (line 304) | class DoubleConv(nn.Module):
    method __init__ (line 307) | def __init__(self, in_channels, out_channels, mid_channels=None):
    method forward (line 316) | def forward(self, x):
  class Down (line 320) | class Down(nn.Module):
    method __init__ (line 323) | def __init__(self, in_channels, out_channels):
    method forward (line 330) | def forward(self, x):
  class Up (line 334) | class Up(nn.Module):
    method __init__ (line 337) | def __init__(self, in_channels, out_channels, bilinear=True):
    method forward (line 349) | def forward(self, x1, x2):
  class OutConv (line 364) | class OutConv(nn.Module):
    method __init__ (line 365) | def __init__(self, in_channels, out_channels):
    method forward (line 369) | def forward(self, x):

FILE: dvs/printer.py
  class Printer (line 3) | class Printer(object):
    method __init__ (line 4) | def __init__(self, *files):
    method open (line 8) | def open(self):
    method close (line 15) | def close(self):
    method write (line 23) | def write(self, obj):
    method flush (line 28) | def flush(self):

FILE: dvs/train.py
  function run_epoch (line 22) | def run_epoch(model, loader, cf, epoch, lr, optimizer=None, is_training=...
  function train (line 150) | def train(args = None):

FILE: dvs/util.py
  function save_train_info (line 11) | def save_train_info(name, checkpoints_dir, cf, model, count, optimizer =...
  function make_dir (line 21) | def make_dir(checkpoints_dir ,cf):
  function get_optimizer (line 38) | def get_optimizer(optimizer, model, init_lr, cf):
  function crop_video (line 45) | def crop_video(in_path, out_path, crop_ratio):
  function norm_flow (line 55) | def norm_flow(flow, h, w):
  class AverageMeter (line 64) | class AverageMeter(object):
    method __init__ (line 65) | def __init__(self):
    method reset (line 68) | def reset(self):
    method update (line 73) | def update(self, val, n=1):

FILE: dvs/warp/rasterizer.py
  function Rasterization (line 10) | def Rasterization(image, grid, get_mesh_only = False):
  function grid_to_triangle (line 73) | def grid_to_triangle(grid):
  function grid_size (line 91) | def grid_size(upper_triangle, lower_triangle, height, width):
  function generate_mesh_grid (line 103) | def generate_mesh_grid(height, width):
  function triangle2mask (line 111) | def triangle2mask(d, height, width): # d: [N x T x 3 x 2]
  function edgefunc (line 144) | def edgefunc(v0, v1, p):

FILE: dvs/warp/read_write.py
  function load_video (line 11) | def load_video(path, save_dir = None, resize = None, length = -1): # N x...
  function video2frame (line 39) | def video2frame(path, resize = None):
  function video2frame_one_seq (line 55) | def video2frame_one_seq(path, save_dir = None, resize = None): # N x H x...
  function save_video (line 79) | def save_video(path,frame_array, fps, size, losses = None, frame_number ...
  function draw_number (line 97) | def draw_number(frame, num, x = 10, y = 10, message = "Frame: "):

FILE: dvs/warp/warping.py
  function warp_video (line 9) | def warp_video(mesh_path, video_path, save_path, losses = None, frame_nu...
  function warpping_rast (line 30) | def warpping_rast(grid_data, frame_array, losses = None):
  function warpping_one_frame_rast (line 37) | def warpping_one_frame_rast(image, grid):