MCPcopy Create free account

hub / github.com/SudeepDasari/RoboNet / functions

Functions384 in github.com/SudeepDasari/RoboNet

↓ 1 callersMethod_get_default_hparams
()
robonet/datasets/base_dataset.py:59
↓ 1 callersMethod_get_graph
(self, graph_type)
robonet/video_prediction/models/base_model.py:21
↓ 1 callersMethod_get_input_targets
(self, DatasetClass, metadata, dataset_hparams)
scripts/examples/create_prediction_gifs.py:54
↓ 1 callersMethod_get_model_class
(self, class_name)
robonet/video_prediction/training/trainable_interface.py:70
↓ 1 callersMethod_get_placeholders
(self)
robonet/datasets/robonet_dataset.py:257
↓ 1 callersMethod_hdf5_generator
(self, sources, sources_metadata, rng, mode)
robonet/datasets/robonet_dataset.py:138
↓ 1 callersMethod_init_dataset
(self)
robonet/datasets/record_dataset.py:26
↓ 1 callersMethod_init_dataset
(self)
robonet/datasets/base_dataset.py:38
↓ 1 callersMethod_init_layers
(self, hparams, inputs, mode, enc_device, dec_device)
robonet/video_prediction/models/graphs/vgg_conv_graph.py:209
↓ 1 callersMethod_init_sources
(self)
robonet/video_prediction/training/trainable_interface.py:129
↓ 1 callersMethod_init_sources
(self)
scripts/examples/create_prediction_gifs.py:91
↓ 1 callersMethod_lazily_initialize
Initialize the graph and session, if this has not yet been done.
robonet/video_prediction/utils/tf_utils.py:542
↓ 1 callersFunction_load_data
(inputs)
robonet/datasets/robonet_dataset.py:12
↓ 1 callersMethod_make_dataloaders
(self, config)
robonet/video_prediction/training/trainable_interface.py:115
↓ 1 callersMethod_make_dataloaders
(self, config)
scripts/examples/create_prediction_gifs.py:77
↓ 1 callersMethod_model_default_hparams
(self)
robonet/video_prediction/models/base_model.py:48
↓ 1 callersMethod_model_fn
(self, inputs, targets, mode)
robonet/video_prediction/models/base_model.py:51
↓ 1 callersMethod_restore
(self, checkpoints)
robonet/video_prediction/training/trainable_interface.py:271
↓ 1 callersMethod_split_files
(self, source_number, metadata)
robonet/datasets/robonet_dataset.py:94
↓ 1 callersFunction_timing_test
(N, loader)
robonet/datasets/robonet_dataset.py:316
↓ 1 callersMethod_wrap_generator
(self, source_files, source_metadata, rng, mode)
robonet/datasets/robonet_dataset.py:135
↓ 1 callersFunctionadd_image_summaries
(outputs, max_outputs=16, collections=None)
robonet/video_prediction/utils/tf_utils.py:182
↓ 1 callersMethodadd_images
(self, ims, txts, links, colspans=None, height=None, width=400)
robonet/video_prediction/utils/html.py:60
↓ 1 callersFunctionadd_scalar_summaries
(losses_or_metrics, collections=None)
robonet/video_prediction/utils/tf_utils.py:204
↓ 1 callersFunctionadd_tensor_summaries
(outputs, max_outputs=16, collections=None)
robonet/video_prediction/utils/tf_utils.py:193
↓ 1 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/layers/dnaflow_rnn_cell.py:47
↓ 1 callersFunctionapply_dna_kernels
Args: image: A 4-D tensor of shape `[batch, in_height, in_width, in_channels]`. kernels: A 6-D of shape `
robonet/video_prediction/layers/dnaflow_rnn_cell.py:12
↓ 1 callersFunctionaverage_and_repeat
:param enc: time, batch, z_dim :param params: :param tlen: length of horizon :return: e in time-major
robonet/video_prediction/models/deterministc_embedding_utils.py:62
↓ 1 callersMethodbuild_graph
(self, inputs, hparams, n_gpus=1, scope_name='graph')
robonet/video_prediction/models/graphs/base_graph.py:5
↓ 1 callersFunctionbytes_feature
(value)
robonet/datasets/util/hdf5_2_records.py:15
↓ 1 callersFunctioncartesian_to_polar
(x, y)
robonet/video_prediction/utils/tf_utils.py:491
↓ 1 callersFunctioncreate_n_layer_encoder
(inputs, nz=8, nef=64, n_laye
robonet/video_prediction/layers/encoder_layers.py:5
↓ 1 callersMethoddefault_hparams
()
robonet/video_prediction/models/graphs/base_graph.py:9
↓ 1 callersMethodget_batch
(self, sess, mode='test')
scripts/examples/create_prediction_gifs.py:145
↓ 1 callersFunctionget_descendants
(tensor, tensors)
robonet/video_prediction/utils/tf_utils.py:101
↓ 1 callersFunctionget_graph_class
(class_name)
robonet/video_prediction/models/graphs/__init__.py:1
↓ 1 callersFunctionget_graph_class
(class_name)
robonet/inverse_model/models/graphs/__init__.py:1
↓ 1 callersFunctionget_metadata_frame
(files)
robonet/datasets/util/metadata_helper.py:133
↓ 1 callersFunctionget_norm_layer
(layer_type)
robonet/video_prediction/ops.py:1050
↓ 1 callersFunctionget_prediction_batches
(dataset, prediction_model, mode='test')
scripts/examples/create_prediction_gifs.py:149
↓ 1 callersMethodget_shuffled_files
(self, rng=None)
robonet/datasets/util/metadata_helper.py:35
↓ 1 callersFunctionget_trainable
(class_name)
robonet/__init__.py:9
↓ 1 callersFunctionget_vgg_dict
(path)
robonet/inverse_model/models/layers/vgg_pretrain.py:6
↓ 1 callersMethodinit_default_hparams
(self, params)
robonet/video_prediction/models/base_model.py:24
↓ 1 callersMethodinitialize_graph
Create the TensorFlow graph needed to compute this operation. This should write ops to the default graph and return `None`.
robonet/video_prediction/utils/tf_utils.py:553
↓ 1 callersFunctioninput_pack
(x)
robonet/video_prediction/functional_ops.py:21
↓ 1 callersFunctionload_actions
(file_pointer, meta_data, hparams)
robonet/datasets/util/hdf5_loader.py:94
↓ 1 callersFunctionload_annotations
(file_pointer, metadata, hparams, cams_to_load)
robonet/datasets/util/hdf5_loader.py:124
↓ 1 callersFunctionload_camera_imgs
(cam_index, file_pointer, file_metadata, target_dims, start_time=0, n_load=None)
robonet/datasets/util/hdf5_loader.py:39
↓ 1 callersFunctionload_states
(file_pointer, meta_data, hparams)
robonet/datasets/util/hdf5_loader.py:78
↓ 1 callersFunctionmain
()
robonet/video_prediction/utils/ffmpeg_gif.py:69
↓ 1 callersFunctionmean_squared_error
L2 distance between tensors true and pred. Args: true: the ground truth image. pred: the predicted image. keep_axis:
robonet/video_prediction/metrics.py:143
↓ 1 callersFunctionmean_squared_error_np
(true, pred, keep_axis=None)
robonet/video_prediction/metrics.py:71
↓ 1 callersFunctionmultiplex_tensors
(dataset, key_name, train_cond=None)
robonet/datasets/util/tensor_multiplexer.py:5
↓ 1 callersFunctiononestep_encoder_fn
(targets, hparams=None)
robonet/video_prediction/models/deterministc_embedding_utils.py:19
↓ 1 callersFunctionpad1d
(inputs, size, strides=(1,), padding='SAME', mode='CONSTANT')
robonet/video_prediction/ops.py:17
↓ 1 callersFunctionplot_buf
(y)
robonet/video_prediction/utils/tf_utils.py:229
↓ 1 callersFunctionplot_summary
Hack that uses pr_curve summaries for 2D plots. Args: x: 1-D tensor with values in increasing order. y: 1-D tensor with stat
robonet/video_prediction/utils/tf_utils.py:265
↓ 1 callersMethodpredict
(self, context_tensors, action_tensors)
robonet/video_prediction/testing/model_evaluation_interface.py:90
↓ 1 callersMethodpredict
(self, start_image, goal_image, context_actions=None, context_frames=None)
robonet/inverse_model/testing/action_inference_interface.py:67
↓ 1 callersFunctionrender_dist
(dist)
robonet/video_prediction/training/util.py:36
↓ 1 callersFunctionsave_gif
To generate a gif from image files, first generate palette from images and then generate the gif from the images and the palette. ffmpeg
robonet/video_prediction/utils/ffmpeg_gif.py:6
↓ 1 callersFunctionsave_record
(filename, trajs)
robonet/datasets/util/hdf5_2_records.py:19
↓ 1 callersMethodschedule_sample
(self, T, B, hparams)
robonet/inverse_model/models/graphs/lstm_baseline.py:139
↓ 1 callersFunctionsigmoid_kl_with_logits
(logits, targets)
robonet/video_prediction/ops.py:1007
↓ 1 callersFunctionsplit_model_inference
we use separate trajectories for the encoder than from the ones used for prediction training :param inputs: dict with tensors in *time-major*
robonet/video_prediction/models/deterministc_embedding_utils.py:33
↓ 1 callersFunctionstructural_similarity_finn
(X, Y, keep_axis=None)
robonet/video_prediction/metrics.py:247
↓ 1 callersFunctionstructural_similarity_finn_np
(true, pred, keep_axis=None)
robonet/video_prediction/metrics.py:102
↓ 1 callersFunctionstructural_similarity_finn_official
(img1, img2, cs_map=False)
robonet/video_prediction/metrics.py:375
↓ 1 callersFunctionstructural_similarity_scikit
(true, pred, keep_axis=None)
robonet/video_prediction/metrics.py:243
↓ 1 callersFunctionstructural_similarity_scikit_np
(true, pred, keep_axis=None)
robonet/video_prediction/metrics.py:98
↓ 1 callersFunctiontensor_to_clip
(tensor)
robonet/video_prediction/utils/tf_utils.py:141
↓ 1 callersFunctiontensor_to_image_batch
(tensor)
robonet/video_prediction/utils/tf_utils.py:157
↓ 1 callersFunctiontest_metrics_equivalence
()
robonet/video_prediction/metrics.py:479
↓ 1 callersFunctiontest_ssim
()
robonet/video_prediction/metrics.py:403
↓ 1 callersFunctiontest_ssim_broadcasting
()
robonet/video_prediction/metrics.py:464
↓ 1 callersFunctiontest_ssim_finn
()
robonet/video_prediction/metrics.py:439
↓ 1 callersFunctiontest_ssim_scikit
()
robonet/video_prediction/metrics.py:421
↓ 1 callersFunctionupsample2d_v2
Possibly less computationally efficient but more memory efficent than upsampled2d.
robonet/video_prediction/ops.py:619
↓ 1 callersFunctionvgg_cosine_similarity
(image0, image1, keep_axis=None)
robonet/video_prediction/metrics.py:275
↓ 1 callersFunctionvgg_cosine_similarity_np
(image0, image1, keep_axis=None)
robonet/video_prediction/metrics.py:348
↓ 1 callersFunctionvgg_preprocess_images
:param image_tensor: float 32 array of Batch x Height x Width x Channel immages (range 0 - 1) :return: pre-processed images (ready to input t
robonet/inverse_model/models/layers/vgg_pretrain.py:10
↓ 1 callersFunctionwrap_host
(summary_dir, summary_queue_len, image_summary_freq, fn)
robonet/video_prediction/models/deterministic_generator.py:27
Method__call__
(self, *args, **kwargs)
robonet/video_prediction/utils/tf_utils.py:569
Method__call__
(self, context_tensors, action_tensors)
robonet/video_prediction/testing/model_evaluation_interface.py:156
Method__call__
(self, input_img, training=True)
robonet/inverse_model/models/graphs/lstm_baseline.py:19
Method__call__
(self, start_image, goal_image, context_actions=None, context_frames=None)
robonet/inverse_model/testing/action_inference_interface.py:76
Method__contains__
(self, item)
robonet/datasets/robonet_dataset.py:108
Method__contains__
(self, item)
robonet/datasets/record_dataset.py:132
Method__contains__
(self, item)
robonet/datasets/base_dataset.py:80
Method__contains__
(self, item)
robonet/datasets/util/metadata_helper.py:50
Method__eq__
(self, other)
robonet/datasets/util/metadata_helper.py:59
Method__ge__
(self, other)
robonet/datasets/util/metadata_helper.py:74
Method__getitem__
(self, item)
robonet/datasets/base_dataset.py:71
Method__getitem__
(self, key)
robonet/datasets/util/tensor_multiplexer.py:40
Method__getitem__
(self, arg)
robonet/datasets/util/metadata_helper.py:47
Method__gt__
(self, other)
robonet/datasets/util/metadata_helper.py:71
Method__init__
(self, batch_size, dataset_files_or_metadata, hparams=dict())
robonet/datasets/robonet_dataset.py:23
Method__init__
(self, dataset_batches, dataset_paths, hparams=dict())
robonet/datasets/record_dataset.py:13
Method__init__
(self, batch_size, dataset_files_or_metadata, hparams=dict())
robonet/datasets/base_dataset.py:12
← previousnext →101–200 of 384, ranked by callers