MCPcopy Create free account

hub / github.com/danqu130/DCEIFlow / functions

Functions210 in github.com/danqu130/DCEIFlow

⨍ Functions 210◇ Types & classes 43
↓ 10 callersMethodlog_info
(self, info, subname=None)
logger.py:153
↓ 8 callersFunctionbuild_module
(module_path, module_name)
utils/utils.py:73
↓ 8 callersMethodupdate
(self)
main.py:41
↓ 7 callersMethod__init__
(self, hidden_dim=128, input_dim=192+128)
core/decoder/with_event_updater.py:18
↓ 6 callersMethod__init__
(self, hidden_dim=128, input_dim=192+128)
core/decoder/raft_updater.py:18
↓ 6 callersFunctioncoords_grid
(batch, ht, wd)
utils/sample_utils.py:55
↓ 5 callersMethodclose
(self)
logger.py:171
↓ 4 callersMethodinit_logger
(self, name=None)
logger.py:113
↓ 4 callersMethodload
(self, path, only_model=True)
trainer.py:116
↓ 3 callersMethod__init__
(self, input_dim=3, output_dim=128, norm_fn='batch', dropout=0.0)
core/backbone/raft_encoder.py:126
↓ 3 callersMethod_make_layer
(self, dim, stride=1)
core/backbone/raft_encoder.py:167
↓ 3 callersMethod_make_layer
(self, dim, stride=1)
core/backbone/raft_encoder.py:244
↓ 3 callersMethodclear
(self)
core/metric/Combine.py:24
↓ 3 callersFunctioneventsToVoxel
(events, num_bins=5, height=None, width=None, event_polarity=False, temporal_bilinear=True)
utils/event_uitls.py:204
↓ 3 callersFunctionevents_to_voxel_torch
Turn set of events to a voxel grid tensor, using temporal bilinear interpolation Parameters ---------- xs : list of event x coordinat
utils/event_uitls.py:91
↓ 3 callersMethodinit_writer
(self)
logger.py:110
↓ 3 callersFunctionprop_flow
(x_flow, y_flow, x_indices, y_indices, x_mask, y_mask, scale_factor=1.0)
utils/datasets/MVSEC_utils.py:52
↓ 3 callersMethodpush
(self, metrics, group=None, last=True)
logger.py:66
↓ 3 callersFunctionreadDenseFlow
(file_name)
utils/file_io.py:96
↓ 3 callersFunctionupflow8
(flow, mode='bilinear')
utils/sample_utils.py:66
↓ 2 callersMethod__init__
(self, enabled)
core/DCEIFlow.py:19
↓ 2 callersFunctionbilinear_sampler
Wrapper for grid_sample, uses pixel coordinates
utils/sample_utils.py:38
↓ 2 callersFunctionbinary_search_torch_tensor
Binary search sorted pytorch tensor
utils/event_uitls.py:22
↓ 2 callersMethodcompute
(self, flow_preds, fmap2_gt, fmap2_pseudo, flow_gt, valid_original)
core/loss/L1Loss.py:32
↓ 2 callersFunctioncount_all_parameters
(model)
utils/utils.py:69
↓ 2 callersFunctioncount_parameters
(model)
utils/utils.py:65
↓ 2 callersFunctionensure_folder
(path)
utils/utils.py:60
↓ 2 callersFunctionepe_f1
(flow_pred, flow_gt, valid_gt=None)
core/metric/EPE.py:37
↓ 2 callersFunctionevaluates
(args, model, datasets, names, metric_fun, logger=None)
evaluate.py:35
↓ 2 callersFunctionevents_to_image_torch
Method to turn event tensor to image. Allows for bilinear interpolation. :param xs: tensor of x coords of events :param ys: tenso
utils/event_uitls.py:42
↓ 2 callersFunctionfetch_augmentor
(is_event=True, is_sparse=False, aug_params=None)
utils/augmentor/__init__.py:7
↓ 2 callersMethodinitialize_flow
Flow is represented as difference between two coordinate grids flow = coords1 - coords0
core/DCEIFlow.py:83
↓ 2 callersMethodlog_debug
(self, debug, subname=None)
logger.py:162
↓ 2 callersMethodopen_hdf5
(self)
utils/datasets/MVSEC.py:125
↓ 2 callersMethodpad
(self, *inputs)
utils/utils.py:43
↓ 2 callersFunctionread_event_h5
(path)
utils/file_io.py:109
↓ 2 callersFunctionread_gen
(file_name)
utils/file_io.py:85
↓ 2 callersMethodresize_sparse_flow_map
(self, flow, valid, fx=1.0, fy=1.0)
utils/augmentor/image_augmentor.py:184
↓ 2 callersMethodresizeflow_tosize
(self, flow, new_size, mode='bilinear')
core/loss/L1Loss.py:23
↓ 2 callersFunctionsetup_seed
(seed)
utils/utils.py:12
↓ 2 callersMethodstep
(self, optimizer)
main.py:39
↓ 2 callersMethodsummary
(self, index)
logger.py:83
↓ 2 callersFunctiontrain
(local_rank, args)
main.py:45
↓ 2 callersMethodunpad
(self, x)
utils/utils.py:46
↓ 2 callersMethodupsample_flow
Upsample flow field [H/8, W/8, 2] -> [H, W, 2] using convex combination
core/DCEIFlow.py:92
↓ 1 callersMethod_clear_summary
(self)
logger.py:60
↓ 1 callersMethod_log_summary
(self, index)
logger.py:35
↓ 1 callersMethod_print
(self)
argparser.py:67
↓ 1 callersMethod_write_summary
(self, index)
logger.py:50
↓ 1 callersMethodbar_update
(self, loss)
trainer.py:202
↓ 1 callersMethodcal
(self, flow_pred, flow_gt, flow_valid, event_valid, name)
core/metric/EPE.py:74
↓ 1 callersMethodcalculate
(self, output, target, name=None)
core/metric/Combine.py:27
↓ 1 callersMethodcolor_transform
Photometric augmentation
utils/augmentor/event_augmentor.py:83
↓ 1 callersMethodcolor_transform
Photometric augmentation
utils/augmentor/event_augmentor.py:193
↓ 1 callersMethodcolor_transform
Photometric augmentation
utils/augmentor/image_augmentor.py:36
↓ 1 callersMethodcolor_transform
(self, img1, img2)
utils/augmentor/image_augmentor.py:164
↓ 1 callersMethodcorr
(fmap1, fmap2)
core/corr/raft_corr.py:56
↓ 1 callersMethodcorr
(fmap1, fmap2)
core/corr/raft_corr.py:109
↓ 1 callersMethodcorr
(self, fmap1, fmap2, coords)
core/corr/raft_corr.py:129
↓ 1 callersFunctionepe
(flow_pred, flow_gt, valid_gt=None)
core/metric/EPE.py:4
↓ 1 callersMethoderaser_transform
Occlusion augmentation
utils/augmentor/image_augmentor.py:55
↓ 1 callersMethoderaser_transform
(self, img1, img2)
utils/augmentor/image_augmentor.py:171
↓ 1 callersFunctionevaluate
(args, model, dataloader, name, metric_fun, logger=None)
evaluate.py:64
↓ 1 callersFunctioneventsToVoxelTorch
(events, num_bins=5, height=None, width=None, event_polarity=False, temporal_bilinear=True)
utils/event_uitls.py:208
↓ 1 callersFunctioneventsToXYTP
(events, process=False)
utils/event_uitls.py:186
↓ 1 callersFunctionevents_to_neg_pos_voxel_torch
Turn set of events to a voxel grid tensor, using temporal bilinear interpolation. Positive and negative events are put into separate voxel gr
utils/event_uitls.py:144
↓ 1 callersFunctionfetch_dataset
Create the data loader for the corresponding training set
utils/datasets/__init__.py:4
↓ 1 callersFunctionfetch_test_dataset
Create the torch Dataset for the corresponding testing set / name
utils/datasets/__init__.py:24
↓ 1 callersMethodfetch_valids
(self)
utils/datasets/FlyingChairs2.py:156
↓ 1 callersMethodforward
Loss function defined over sequence of flow predictions
core/loss/L1Loss.py:86
↓ 1 callersFunctiongenerate_corresponding_gt_flow
(flows, flows_ts, start_time,
utils/datasets/MVSEC_utils.py:96
↓ 1 callersMethodget_raw_events
(self, index)
utils/datasets/FlyingChairs2.py:326
↓ 1 callersMethodinitialize_flow
Flow is represented as difference between two coordinate grids flow = coords1 - coords0
core/RAFT.py:64
↓ 1 callersFunctioninterpolate_to_image
Accumulate x and y coords to an image using bilinear interpolation
utils/event_uitls.py:11
↓ 1 callersMethodis_init_logger
(self)
logger.py:182
↓ 1 callersMethodis_init_writer
(self)
logger.py:179
↓ 1 callersMethodload_data_by_index
(self, index)
utils/datasets/FlyingChairs2.py:212
↓ 1 callersMethodpad_batch
(self, batch)
utils/utils.py:34
↓ 1 callersMethodparser
(self)
argparser.py:77
↓ 1 callersFunctionreadFlow
Read .flo file in Middlebury format
utils/file_io.py:17
↓ 1 callersFunctionreadPFM
(file)
utils/file_io.py:39
↓ 1 callersFunctionreduce_list
(lists, nprocs)
trainer.py:12
↓ 1 callersFunctionreduce_list
(lists, nprocs)
evaluate.py:18
↓ 1 callersMethodrun_epoch
(self, dataloader)
trainer.py:163
↓ 1 callersMethodscale
(self, loss)
main.py:35
↓ 1 callersMethodspatial_transform
(self, event, img1, img2, flow, flow10=None, occ=None, occ10=None, event_r=None)
utils/augmentor/event_augmentor.py:102
↓ 1 callersMethodspatial_transform
(self, event, img1, img2, flow, valid=None, flow10=None, valid10=None)
utils/augmentor/event_augmentor.py:212
↓ 1 callersMethodspatial_transform
(self, img1, img2, flow, flow10)
utils/augmentor/image_augmentor.py:70
↓ 1 callersMethodspatial_transform
(self, img1, img2, flow, valid, flow10=None, valid10=None)
utils/augmentor/image_augmentor.py:218
↓ 1 callersMethodstore
(self, path, name, index=None)
trainer.py:150
↓ 1 callersMethodunscale_
(self, optimizer)
main.py:37
↓ 1 callersMethodupsample_flow
Upsample flow field [H/8, W/8, 2] -> [H, W, 2] using convex combination
core/RAFT.py:73
↓ 1 callersMethodweight_fix
(self, way, refer_dict=None)
trainer.py:47
↓ 1 callersMethodwrite_dict
(self, index, results, group=None)
logger.py:88
Method__call__
(self, event, img1, img2, flow, flow10=None, occ=None, occ10=None, event_r=None)
utils/augmentor/event_augmentor.py:151
Method__call__
(self, event, img1, img2, flow, valid=None, flow10=None, valid10=None)
utils/augmentor/event_augmentor.py:255
Method__call__
(self, img1, img2, flow, flow10=None)
utils/augmentor/image_augmentor.py:129
Method__call__
(self, img1, img2, flow, valid, flow10=None, valid10=None)
utils/augmentor/image_augmentor.py:270
Method__call__
(self, coords)
core/corr/raft_corr.py:31
Method__call__
(self, coords)
core/corr/raft_corr.py:84
next →1–100 of 210, ranked by callers