MCPcopy Create free account

hub / github.com/SudeepDasari/RoboNet / functions

Functions384 in github.com/SudeepDasari/RoboNet

Methodfiles
(self)
robonet/datasets/util/metadata_helper.py:32
Functionflow_to_rgb
The last axis should have dimension 2, for x and y values.
robonet/video_prediction/utils/tf_utils.py:488
Functionfn1
(**kwargs)
robonet/video_prediction/models/deterministic_generator.py:28
Functionfoldl
Same as tf.foldl but with support for a possibly nested sequence of tensors.
robonet/video_prediction/functional_ops.py:11
Methodframe
(self)
robonet/datasets/util/metadata_helper.py:28
Functionfused_instance_norm
Functional interface for the instance normalization layer. Reference: https://arxiv.org/abs/1607.08022. "Instance Normalization: The Missing I
robonet/video_prediction/layers/normalization.py:34
Functiongan_loss
(logits, labels, gan_loss_type)
robonet/video_prediction/losses.py:14
Functiongaussian2
(size, sigma)
robonet/video_prediction/metrics.py:362
Functionget_checkpoint_restore_saver
(checkpoint, skip_global_step=False, restore_to_checkpoint_mapping=None, rest
robonet/video_prediction/utils/tf_utils.py:428
Functionget_downsample_layer
(layer_type)
robonet/video_prediction/ops.py:1075
Methodget_image_dir
(self)
robonet/video_prediction/utils/html.py:24
Functionget_mp4
(traj_id, cam_id)
scripts/visualize_dataset.py:43
Functionget_trainable
(class_name)
robonet/video_prediction/training/__init__.py:4
Functionget_trainable
(name)
robonet/inverse_model/training/__init__.py:1
Functionget_upsample_layer
(layer_type)
robonet/video_prediction/ops.py:1063
Methodhorizon
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:198
Methodhorizon
(self)
robonet/inverse_model/testing/action_inference_interface.py:105
Functionhost_summary_fn
(summary_dir, summary_queue_len, image_summary_freq, **summary_dict)
robonet/video_prediction/models/deterministic_generator.py:14
Methodhparams
(self)
robonet/datasets/base_dataset.py:88
Functionimage_warp
Performs a backward warp of an image using the predicted flow. Args: im: Batch of images. [num_batch, height, width, channels] fl
robonet/video_prediction/flow_ops.py:4
Methodimg_size
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:206
Functionindex
()
scripts/visualize_dataset.py:19
Methodinitialize_graph
(self)
robonet/video_prediction/metrics.py:325
Functioninstancenorm
(input)
robonet/video_prediction/ops.py:918
Methoditeration
(self)
robonet/video_prediction/training/trainable_interface.py:286
Functionkl_loss
(mu, log_sigma_sq)
robonet/video_prediction/losses.py:42
Functionkronecker_local2d
2-D locally connected operation with filters represented as a kronecker product of smaller filters Args: inputs: A 4-D tensor of sha
robonet/video_prediction/ops.py:350
Functionl1_loss
(pred, target)
robonet/video_prediction/losses.py:6
Functionl2_loss
(pred, target)
robonet/video_prediction/losses.py:10
Functionload_metadata_dict
(fname)
robonet/datasets/util/metadata_helper.py:84
Functionlocal2d
2-D locally connected operation. Works similarly to 2-D convolution except that the weights are unshared, that is, a different set of filter
robonet/video_prediction/ops.py:158
Functionlocal_device_setter
(num_devices=1, ps_device_type='cpu', worker_device='/cpu:0',
robonet/video_prediction/utils/tf_utils.py:34
Methodmake_dataloaders
(self, config)
robonet/video_prediction/training/finetuning_trainable_interface.py:15
Functionmaybe_pad_or_slice
(tensor, desired_length)
robonet/video_prediction/utils/tf_utils.py:130
Methodmodel_hparams
(self)
robonet/video_prediction/models/base_model.py:63
Methodmodes
(self)
robonet/datasets/record_dataset.py:136
Methodmodes
(self)
robonet/datasets/base_dataset.py:96
Methodmodes
(self)
robonet/datasets/variants/annotation_benchmark_dataset.py:42
Methodmodes
(self)
robonet/datasets/util/tensor_multiplexer.py:61
Methodn_cam
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:202
Methodn_context
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:194
Methodnum_examples_per_epoch
(self)
robonet/datasets/record_dataset.py:140
Methodnum_examples_per_epoch
(self)
robonet/datasets/base_dataset.py:92
Methodon_result
(self, result)
robonet/video_prediction/training/ray_util/gif_logger.py:39
Methodoutput_size
(self)
robonet/video_prediction/rnn_ops.py:97
Methodoutput_size
(self)
robonet/video_prediction/rnn_ops.py:186
Methodoutput_size
(self)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:250
Functionparse_tpu_config
Configures custom yaml loading behavior and parses config file
robonet/yaml_util.py:37
Functionparse_tune_config
Configures custom yaml loading behavior and parses config file
robonet/yaml_util.py:4
Functionpeak_signal_to_noise_ratio
Image quality metric based on maximal signal power vs. power of the noise. Args: true: the ground truth image. pred: the pre
robonet/video_prediction/metrics.py:122
Functionpeak_signal_to_noise_ratio_np
(true, pred, keep_axis=None)
robonet/video_prediction/metrics.py:63
Functionpixel_distribution
(pos, height, width)
robonet/video_prediction/utils/tf_utils.py:462
Functionpower_iteration
(i, u_i, v_i)
robonet/video_prediction/ops.py:1030
Methodprimary_mode
(self)
robonet/datasets/base_dataset.py:100
Functionprint_loss_info
(losses, inputs, outputs, targets)
robonet/video_prediction/utils/tf_utils.py:100
Functionreduce_tensors
(structures, shallow=False)
robonet/video_prediction/utils/tf_utils.py:413
Functionreplace_read_ops
Replaces read ops of each variable in `vars` with new read ops obtained from `read_value()`, thus forcing to read the most up-to-date values
robonet/video_prediction/utils/tf_utils.py:65
Methodrestore
(self)
robonet/inverse_model/testing/action_inference_interface.py:82
Methodscope_name
(self)
robonet/video_prediction/models/base_model.py:55
Methodsdim
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:214
Functionsearch_constructor
(loader, node)
robonet/yaml_util.py:10
Functionseparable_local2d
2-D locally connected operation with separable filters. Note that, unlike tf.nn.separable_conv2d, this is spatial separability between dimen
robonet/video_prediction/ops.py:249
Methodsequence_length
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:190
Methodset_session
(self, sess)
robonet/video_prediction/testing/model_evaluation_interface.py:159
Methodset_session
(self, sess)
robonet/inverse_model/testing/action_inference_interface.py:79
Methodstate_size
(self)
robonet/video_prediction/rnn_ops.py:101
Methodstate_size
(self)
robonet/video_prediction/rnn_ops.py:182
Methodstate_size
(self)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:254
Methodtrain_val_filter
(self, train_metadata, val_metadata)
robonet/datasets/variants/val_filter_dataset_variants.py:51
Methodtrain_val_filter
(self, train_metadata, val_metadata)
robonet/datasets/variants/val_filter_dataset_variants.py:67
Functiontraj_page
(traj_id)
scripts/visualize_dataset.py:30
Functiontranspose_batch_time
(x)
robonet/video_prediction/utils/tf_utils.py:123
Functiontrial_str_creator
(trial)
scripts/train_model.py:10
Functionupsample2d
(inputs, strides, padding='SAME', upsample_mode='bilinear')
robonet/video_prediction/ops.py:601
Functionupsample_conv2d
Upsamples the inputs by a factor using bilinear interpolation and the performs conv2d on the upsampled input. This function is more computati
robonet/video_prediction/ops.py:641
Functionupsample_conv2d_v2
(inputs, filters, kernel_size, strides=(1, 1), padding='SAME', kernel=None, use_bias=Tr
robonet/video_prediction/ops.py:720
Methodvars
(self)
robonet/video_prediction/models/graphs/deterministic_graph.py:36
Methodvars
(self)
robonet/video_prediction/models/graphs/base_graph.py:17
Functionvgg16
rgb_image: 4-D tensor with pixel intensities between 0 and 1.
robonet/video_prediction/layers/vgg_network.py:38
Functionvgg19
rgb_image: 4-D tensor with pixel intensities between 0 and 1.
robonet/video_prediction/layers/vgg_network.py:105
Functionvgg_assign_from_values_fn
(model='vgg16', var_name_prefix='vgg/', var_name_k
robonet/video_prediction/layers/vgg_network.py:5
Functionvgg_cosine_distance
(image0, image1, keep_axis=None)
robonet/video_prediction/metrics.py:291
Functionvgg_cosine_distance_np
(image0, image1, keep_axis=None)
robonet/video_prediction/metrics.py:358
Methodzero_state
(self, batch_size, dtype)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:257
← previous301–384 of 384, ranked by callers