MCPcopy Create free account

hub / github.com/SudeepDasari/RoboNet / functions

Functions384 in github.com/SudeepDasari/RoboNet

↓ 49 callersMethodconcat
(tensors, axis)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:321
↓ 27 callersMethodget
(self, key, mode='train')
robonet/datasets/base_dataset.py:66
↓ 22 callersFunctionconv2d
2-D convolution. Args: inputs: A 4-D tensor of shape `[batch, in_height, in_width, in_channels]`. kernel: A 4-D
robonet/video_prediction/ops.py:492
↓ 18 callersMethodrun
(self, image)
robonet/video_prediction/metrics.py:331
↓ 11 callersFunction_cast_down
(tensor, hparams)
robonet/video_prediction/models/graphs/vgg_conv_graph.py:42
↓ 11 callersMethodrun
Evaluate the ops with the given input. When this function is called, the default session will have the graph defined by a previous cal
robonet/video_prediction/utils/tf_utils.py:559
↓ 11 callersFunctiontile_concat
Like concat except that first tiles the broadcastable dimensions if necessary
robonet/video_prediction/ops.py:966
↓ 10 callersMethodkeys
(self)
robonet/datasets/util/metadata_helper.py:77
↓ 9 callersFunction_axis
(keep_axis, ndims)
robonet/video_prediction/metrics.py:49
↓ 9 callersFunctionload_metadata
(files)
robonet/datasets/util/metadata_helper.py:165
↓ 9 callersFunctionpad2d
Pads a 4-D tensor according to the convolution padding algorithm. Convolution with a padding scheme conv2d(..., padding=padding)
robonet/video_prediction/ops.py:127
↓ 8 callersFunctiondense
(inputs, units, use_spectral_norm=False)
robonet/video_prediction/ops.py:5
↓ 8 callersMethodget_feed_dict
(self, mode)
robonet/datasets/util/tensor_multiplexer.py:47
↓ 7 callersFunction_cast_up
(tensor)
robonet/video_prediction/models/graphs/vgg_conv_graph.py:50
↓ 7 callersFunctionvgg_conv
(vgg_dict, bottom, name)
robonet/inverse_model/models/layers/vgg_pretrain.py:25
↓ 6 callersMethod_norm
(self, inputs, scope)
robonet/video_prediction/rnn_ops.py:104
↓ 6 callersFunctionload_data
(f_name, file_metadata, hparams, rng=None)
robonet/datasets/util/hdf5_loader.py:145
↓ 5 callersFunction_as_name_scope_map
(values)
robonet/video_prediction/utils/tf_utils.py:173
↓ 4 callersMethod_conv_rnn_func
(self, inputs, state, filters)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:264
↓ 4 callersMethod_norm
(self, inputs, scope, bias_initializer)
robonet/video_prediction/rnn_ops.py:189
↓ 4 callersFunctionapply_cdna_kernels
Args: image: A 4-D tensor of shape `[batch, in_height, in_width, in_channels]`. kernels: A 4-D of shape `
robonet/video_prediction/models/graphs/vgg_conv_graph.py:9
↓ 4 callersFunctionapply_flows
(image, flows)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:109
↓ 4 callersFunctionapply_kernels
Args: image: A 4-D tensor of shape `[batch, in_height, in_width, in_channels]`. kernels: A 4-D or 6-D tensor of shape
robonet/video_prediction/layers/dnaflow_rnn_cell.py:80
↓ 4 callersFunctionflatten
Caffe-style flatten. Args: inputs: An N-D tensor. axis: The first axis to flatten: all preceding axes are retained in the ou
robonet/video_prediction/ops.py:934
↓ 4 callersFunctionget_dataset_class
(name)
robonet/datasets/__init__.py:4
↓ 4 callersMethodget_file_metadata
(self, fname)
robonet/datasets/util/metadata_helper.py:18
↓ 4 callersFunctionidentity_kernel
(kernel_size)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:122
↓ 4 callersFunctionsplit_train_val_test
(metadata, splits=None, train_ex=None, rng=None)
robonet/datasets/util/dataset_utils.py:23
↓ 4 callersFunctionstructural_similarity
Structural SIMilarity (SSIM) index between two images Args: X: A tensor of shape `[..., in_height, in_width, in_channels]`.
robonet/video_prediction/metrics.py:161
↓ 4 callersFunctionstructural_similarity_np
(true, pred, K1=0.01, K2=0.03, sigma=1.5, win_size=None, data_range=1.0, gaussian
robonet/video_prediction/metrics.py:76
↓ 4 callersFunctionvgg_pool
(bottom, name)
robonet/inverse_model/models/layers/vgg_pretrain.py:38
↓ 4 callersFunctionwith_flat_batch
(flat_batch_fn)
robonet/video_prediction/utils/tf_utils.py:575
↓ 3 callersMethod_filter_metadata
(self, metadata_list)
robonet/video_prediction/training/data_filter.py:34
↓ 3 callersMethod_get_default_hparams
(parent_hparams=None)
robonet/datasets/variants/val_filter_dataset_variants.py:45
↓ 3 callersFunctioncolor_augment
(image, noise_range=0.2)
robonet/datasets/util/dataset_utils.py:6
↓ 3 callersMethodconcat
(tensors, axis)
robonet/video_prediction/layers/deterministic_embedding_rnn_cell.py:44
↓ 3 callersFunctiondeconv2d
2-D transposed convolution. Notes on padding: The equivalent of transposed convolution with full padding is a convolution with valid
robonet/video_prediction/ops.py:551
↓ 3 callersFunctiondefault_loader_hparams
()
robonet/datasets/util/hdf5_loader.py:24
↓ 3 callersFunctionencode_gif
(images, fps)
robonet/video_prediction/utils/ffmpeg_gif.py:44
↓ 3 callersFunctionget_bilinear_kernel
(strides)
robonet/video_prediction/ops.py:590
↓ 3 callersFunctionget_model
(class_name)
robonet/video_prediction/models/__init__.py:4
↓ 3 callersMethodmodel_fn
(self, features, labels, mode, params)
robonet/video_prediction/models/base_model.py:44
↓ 3 callersFunctionpad2d_paddings
Computes the paddings for a 4-D tensor according to the convolution padding algorithm. See pad2d. Reference: https://www.tensor
robonet/video_prediction/ops.py:69
↓ 3 callersFunctionpad_and_concat
(real_frames, pred_frames, pad_amount)
robonet/video_prediction/training/util.py:22
↓ 3 callersFunctionpool2d
(inputs, pool_size, strides=(1, 1), padding='SAME', pool_mode='avg')
robonet/video_prediction/ops.py:778
↓ 3 callersFunctionspectral_normed_weight
(W, u=None, num_iters=1)
robonet/video_prediction/ops.py:1018
↓ 3 callersFunctionstbmajor
swap time-batch major :param ten: npy tenosr :return:
robonet/video_prediction/training/util.py:5
↓ 3 callersMethodtrain_val_filter
:param metadata: :return: train_metadata, val_metadata
robonet/datasets/variants/val_filter_dataset_variants.py:33
↓ 2 callersFunction_binarize
(actions, pivots)
robonet/inverse_model/models/discretized_inverse_model.py:12
↓ 2 callersMethod_conv2d
(self, inputs, output_filters, bias_initializer)
robonet/video_prediction/rnn_ops.py:200
↓ 2 callersMethod_default_hparams
(self)
robonet/video_prediction/training/data_filter.py:29
↓ 2 callersMethod_default_source_hparams
(self)
robonet/video_prediction/training/trainable_interface.py:122
↓ 2 callersMethod_default_source_hparams
(self)
scripts/examples/create_prediction_gifs.py:84
↓ 2 callersMethod_get_input_targets
(self, DatasetClass, metadata, dataset_hparams)
robonet/video_prediction/training/trainable_interface.py:92
↓ 2 callersMethod_init_rng
(self)
robonet/datasets/base_dataset.py:41
↓ 2 callersMethod_tf_log
(self, result)
robonet/video_prediction/training/trainable_interface.py:289
↓ 2 callersMethodadd_table
(self, border=1)
robonet/video_prediction/utils/html.py:39
↓ 2 callersMethodbuild_feed_dict
(self, mode)
robonet/datasets/robonet_dataset.py:281
↓ 2 callersMethodbuild_feed_dict
(self, mode)
robonet/datasets/base_dataset.py:103
↓ 2 callersMethodbuild_graph
(self, mode, inputs, hparams, scope_name='flow_generator')
robonet/inverse_model/models/graphs/lstm_baseline.py:51
↓ 2 callersFunctioncenter_slice
(k)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:126
↓ 2 callersFunctionconstruct_image_tile
(tensor)
robonet/video_prediction/utils/encode_img.py:7
↓ 2 callersFunctioncosine_similarity
(tensor0, tensor1, keep_axis=None)
robonet/video_prediction/metrics.py:258
↓ 2 callersFunctioncosine_similarity_np
(tensor0, tensor1, keep_axis=None)
robonet/video_prediction/metrics.py:300
↓ 2 callersFunctionexp_pix
(tensor)
robonet/video_prediction/metrics.py:33
↓ 2 callersFunctionexp_square_pix
(tensor)
robonet/video_prediction/metrics.py:38
↓ 2 callersFunctionexpected_pixel_distribution_np
(pix_distrib)
robonet/video_prediction/metrics.py:108
↓ 2 callersFunctionfloat_feature
(value)
robonet/datasets/util/hdf5_2_records.py:11
↓ 2 callersMethodflush
(self)
robonet/video_prediction/training/ray_util/gif_logger.py:23
↓ 2 callersFunctionfn
(x, *args, **kwargs)
robonet/video_prediction/utils/tf_utils.py:576
↓ 2 callersFunctionfspecial_gauss
(size, sigma)
robonet/video_prediction/metrics.py:369
↓ 2 callersFunctionget_inds
(ind, lengths)
robonet/video_prediction/ops.py:412
↓ 2 callersFunctionget_models
(class_name)
robonet/inverse_model/models/__init__.py:1
↓ 2 callersFunctionl2normalize
(v, eps=1e-12)
robonet/video_prediction/ops.py:1027
↓ 2 callersFunctionlrelu
Leaky ReLU activation function Reference: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/ops/nn_ops.py
robonet/video_prediction/ops.py:893
↓ 2 callersFunctionnormalize_tensor
(tensor, eps=1e-10)
robonet/video_prediction/metrics.py:253
↓ 2 callersFunctionnormalize_tensor_np
(tensor, eps=1e-10)
robonet/video_prediction/metrics.py:295
↓ 2 callersFunctionoutput_pack
(x)
robonet/video_prediction/functional_ops.py:31
↓ 2 callersFunctionpad
(real_frames, pad_amount)
robonet/video_prediction/training/util.py:14
↓ 2 callersMethodrestore
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:162
↓ 2 callersMethodsave
(self)
robonet/video_prediction/utils/html.py:86
↓ 2 callersMethodselect_objects
(self, obj_class_name)
robonet/datasets/util/metadata_helper.py:22
↓ 2 callersFunctionsplit
(inputs, bs, sbs)
robonet/video_prediction/models/deterministc_embedding_utils.py:40
↓ 1 callersMethod__init__
(self, input_shape, filters, kernel_size, activation_fn=tf.tanh, normalizer_
robonet/video_prediction/rnn_ops.py:167
↓ 1 callersMethod_build_dataset
(self, files, mode, dataset_metadata, batch_size)
robonet/datasets/record_dataset.py:72
↓ 1 callersMethod_build_input_targets
(self)
robonet/inverse_model/testing/action_inference_interface.py:54
↓ 1 callersMethod_build_inputs
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:51
↓ 1 callersMethod_conv2d
(self, inputs)
robonet/video_prediction/rnn_ops.py:115
↓ 1 callersMethod_default_hparams
(self)
robonet/video_prediction/training/trainable_interface.py:51
↓ 1 callersMethod_default_hparams
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:43
↓ 1 callersMethod_default_hparams
(self)
robonet/inverse_model/testing/action_inference_interface.py:47
↓ 1 callersMethod_default_hparams
(self)
scripts/examples/create_prediction_gifs.py:19
↓ 1 callersMethod_default_scope
(self)
robonet/video_prediction/models/base_model.py:18
↓ 1 callersMethod_extract_hparams
Grabs and (optionally) modifies hparams
robonet/video_prediction/training/trainable_interface.py:73
↓ 1 callersMethod_extract_hparams
Grabs and (optionally) modifies hparams
scripts/examples/create_prediction_gifs.py:37
↓ 1 callersMethod_feed
(self, context_images, context_actions, context_states, context_distributions, input_actions)
robonet/video_prediction/testing/model_evaluation_interface.py:136
↓ 1 callersMethod_get
(self, key, mode)
robonet/datasets/base_dataset.py:55
↓ 1 callersMethod_get_dataset_class
(self, class_name)
robonet/video_prediction/training/trainable_interface.py:67
↓ 1 callersMethod_get_dataset_class
(self, class_name)
scripts/examples/create_prediction_gifs.py:34
↓ 1 callersMethod_get_default_hparams
()
robonet/datasets/record_dataset.py:118
next →1–100 of 384, ranked by callers