Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/ShuweiShao/AF-SfMLearner
/ functions
Functions
158 in github.com/ShuweiShao/AF-SfMLearner
⨍
Functions
158
◇
Types & classes
32
Method
__init__
(self, batch_size, height, width)
layers.py:140
Method
__init__
(self, batch_size, height, width, eps=1e-7)
layers.py:172
Method
__init__
(self, batch_size, height, width, eps=1e-7)
layers.py:198
Method
__init__
Instiantiate the block :param size: size of input to the spatial transformer block :param mode: method of interpolati
layers.py:392
Method
__init__
(self, size, batch_size, height, width, eps=1e-7)
layers.py:434
Method
__init__
(self, size)
layers.py:463
Method
__init__
(self, size, mode='bilinear')
layers.py:487
Method
__init__
(self, size, batch_size)
layers.py:527
Method
__init__
(self)
layers.py:612
Method
__init__
(self, check_num, dataset)
layers.py:631
Method
__init__
(self)
options.py:10
Method
__init__
(self, options)
trainer_stage_two.py:18
Method
__init__
(self, options)
trainer_end_to_end.py:18
Method
__init__
(self, num_ch_enc, scales= range(4), num_output_channels=1, use_skips=True)
networks/depth_decoder.py:18
Method
__init__
(self, num_input_frames)
networks/pose_cnn.py:8
Method
__init__
(self, num_ch_enc, scales = range(4) , num_output_channels=3, use_skips=True)
networks/appearance_flow_decoder.py:12
Method
__init__
(self, num_ch_enc, scales = range(4) , num_output_channels=1, use_skips=True)
networks/depth_decoder_ICRA.py:13
Method
__init__
(self, num_channels, reduction_ratio=16)
networks/depth_decoder_ICRA.py:71
Method
__init__
(self, num_channels)
networks/depth_decoder_ICRA.py:124
Method
__init__
(self, num_channels, reduction_ratio=16)
networks/depth_decoder_ICRA.py:196
Method
__init__
(self, num_ch_enc, num_input_features, num_frames_to_predict_for=None, stride=1)
networks/pose_decoder.py:9
Method
__init__
(self, block, layers, num_classes=1000, num_input_images=1)
networks/resnet_encoder.py:15
Method
__init__
(self, num_ch_enc, scales = range(4) , num_output_channels=2, use_skips=True)
networks/optical_flow_decoder.py:13
Method
__init__
(self, data_path, filenames, height, width
datasets/mono_dataset.py:37
Method
__init__
(self, *args, **kwargs)
datasets/scared_dataset.py:38
Method
__len__
(self)
datasets/mono_dataset.py:106
Method
check_depth
(self)
datasets/scared_dataset.py:24
Function
compute_depth_errors
Computation of error metrics between predicted and ground truth depths
layers.py:369
Function
download_model_if_doesnt_exist
If pretrained kitti model doesn't exist, download and unzip it
utils.py:45
Method
forward
(self, x)
layers.py:113
Method
forward
(self, x)
layers.py:131
Method
forward
(self, depth, inv_K)
layers.py:161
Method
forward
(self, points, K, T)
layers.py:180
Method
forward
(self, points, K, T)
layers.py:206
Method
forward
(self, x, y)
layers.py:295
Method
forward
Push the src and flow through the spatial transform block :param src: the source image :param flow: the output from t
layers.py:409
Method
forward
(self, points, K, T)
layers.py:450
Method
forward
(self, flow, th=0.95)
layers.py:474
Method
forward
(self, flow12, flow21, scale=0.01, bias=0.5)
layers.py:499
Method
forward
(self, flow)
layers.py:539
Method
forward
(self, pred, target)
layers.py:615
Method
forward
(self, match, mask, visualizer=None)
layers.py:665
Method
forward
(self, input_features)
networks/depth_decoder.py:50
Method
forward
(self, out)
networks/pose_cnn.py:30
Method
forward
(self, input_features)
networks/appearance_flow_decoder.py:44
Method
forward
(self, input_features)
networks/depth_decoder_ICRA.py:51
Method
forward
(self, input_tensor)
networks/depth_decoder_ICRA.py:81
Method
forward
(self, x)
networks/depth_decoder_ICRA.py:110
Method
forward
(self, input_tensor)
networks/depth_decoder_ICRA.py:146
Method
forward
(self, input_tensor)
networks/depth_decoder_ICRA.py:204
Method
forward
(self, input_features)
networks/pose_decoder.py:29
Method
forward
(self, input_image)
networks/resnet_encoder.py:81
Method
forward
(self, input_features)
networks/optical_flow_decoder.py:49
Method
get_color
(self, folder, frame_index, side, do_flip)
datasets/scared_dataset.py:28
Method
get_depth
(self, folder, frame_index, side, do_flip)
datasets/scared_dataset.py:48
Function
get_texu_mask
(non_rigid, rigid)
layers.py:546
Function
pil_loader
(path)
datasets/mono_dataset.py:16
Method
val
Validate the model on a single minibatch
trainer_end_to_end.py:626
← previous
101–158 of 158, ranked by callers