Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/SudeepDasari/RoboNet
/ functions
Functions
384 in github.com/SudeepDasari/RoboNet
⨍
Functions
384
◇
Types & classes
40
↳
Endpoints
3
Method
files
(self)
robonet/datasets/util/metadata_helper.py:32
Function
flow_to_rgb
The last axis should have dimension 2, for x and y values.
robonet/video_prediction/utils/tf_utils.py:488
Function
fn1
(**kwargs)
robonet/video_prediction/models/deterministic_generator.py:28
Function
foldl
Same as tf.foldl but with support for a possibly nested sequence of tensors.
robonet/video_prediction/functional_ops.py:11
Method
frame
(self)
robonet/datasets/util/metadata_helper.py:28
Function
fused_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
Function
gan_loss
(logits, labels, gan_loss_type)
robonet/video_prediction/losses.py:14
Function
gaussian2
(size, sigma)
robonet/video_prediction/metrics.py:362
Function
get_checkpoint_restore_saver
(checkpoint, skip_global_step=False, restore_to_checkpoint_mapping=None, rest
robonet/video_prediction/utils/tf_utils.py:428
Function
get_downsample_layer
(layer_type)
robonet/video_prediction/ops.py:1075
Method
get_image_dir
(self)
robonet/video_prediction/utils/html.py:24
Function
get_mp4
(traj_id, cam_id)
scripts/visualize_dataset.py:43
Function
get_trainable
(class_name)
robonet/video_prediction/training/__init__.py:4
Function
get_trainable
(name)
robonet/inverse_model/training/__init__.py:1
Function
get_upsample_layer
(layer_type)
robonet/video_prediction/ops.py:1063
Method
horizon
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:198
Method
horizon
(self)
robonet/inverse_model/testing/action_inference_interface.py:105
Function
host_summary_fn
(summary_dir, summary_queue_len, image_summary_freq, **summary_dict)
robonet/video_prediction/models/deterministic_generator.py:14
Method
hparams
(self)
robonet/datasets/base_dataset.py:88
Function
image_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
Method
img_size
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:206
Function
index
()
scripts/visualize_dataset.py:19
Method
initialize_graph
(self)
robonet/video_prediction/metrics.py:325
Function
instancenorm
(input)
robonet/video_prediction/ops.py:918
Method
iteration
(self)
robonet/video_prediction/training/trainable_interface.py:286
Function
kl_loss
(mu, log_sigma_sq)
robonet/video_prediction/losses.py:42
Function
kronecker_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
Function
l1_loss
(pred, target)
robonet/video_prediction/losses.py:6
Function
l2_loss
(pred, target)
robonet/video_prediction/losses.py:10
Function
load_metadata_dict
(fname)
robonet/datasets/util/metadata_helper.py:84
Function
local2d
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
Function
local_device_setter
(num_devices=1, ps_device_type='cpu', worker_device='/cpu:0',
robonet/video_prediction/utils/tf_utils.py:34
Method
make_dataloaders
(self, config)
robonet/video_prediction/training/finetuning_trainable_interface.py:15
Function
maybe_pad_or_slice
(tensor, desired_length)
robonet/video_prediction/utils/tf_utils.py:130
Method
model_hparams
(self)
robonet/video_prediction/models/base_model.py:63
Method
modes
(self)
robonet/datasets/record_dataset.py:136
Method
modes
(self)
robonet/datasets/base_dataset.py:96
Method
modes
(self)
robonet/datasets/variants/annotation_benchmark_dataset.py:42
Method
modes
(self)
robonet/datasets/util/tensor_multiplexer.py:61
Method
n_cam
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:202
Method
n_context
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:194
Method
num_examples_per_epoch
(self)
robonet/datasets/record_dataset.py:140
Method
num_examples_per_epoch
(self)
robonet/datasets/base_dataset.py:92
Method
on_result
(self, result)
robonet/video_prediction/training/ray_util/gif_logger.py:39
Method
output_size
(self)
robonet/video_prediction/rnn_ops.py:97
Method
output_size
(self)
robonet/video_prediction/rnn_ops.py:186
Method
output_size
(self)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:250
Function
parse_tpu_config
Configures custom yaml loading behavior and parses config file
robonet/yaml_util.py:37
Function
parse_tune_config
Configures custom yaml loading behavior and parses config file
robonet/yaml_util.py:4
Function
peak_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
Function
peak_signal_to_noise_ratio_np
(true, pred, keep_axis=None)
robonet/video_prediction/metrics.py:63
Function
pixel_distribution
(pos, height, width)
robonet/video_prediction/utils/tf_utils.py:462
Function
power_iteration
(i, u_i, v_i)
robonet/video_prediction/ops.py:1030
Method
primary_mode
(self)
robonet/datasets/base_dataset.py:100
Function
print_loss_info
(losses, inputs, outputs, targets)
robonet/video_prediction/utils/tf_utils.py:100
Function
reduce_tensors
(structures, shallow=False)
robonet/video_prediction/utils/tf_utils.py:413
Function
replace_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
Method
restore
(self)
robonet/inverse_model/testing/action_inference_interface.py:82
Method
scope_name
(self)
robonet/video_prediction/models/base_model.py:55
Method
sdim
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:214
Function
search_constructor
(loader, node)
robonet/yaml_util.py:10
Function
separable_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
Method
sequence_length
(self)
robonet/video_prediction/testing/model_evaluation_interface.py:190
Method
set_session
(self, sess)
robonet/video_prediction/testing/model_evaluation_interface.py:159
Method
set_session
(self, sess)
robonet/inverse_model/testing/action_inference_interface.py:79
Method
state_size
(self)
robonet/video_prediction/rnn_ops.py:101
Method
state_size
(self)
robonet/video_prediction/rnn_ops.py:182
Method
state_size
(self)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:254
Method
train_val_filter
(self, train_metadata, val_metadata)
robonet/datasets/variants/val_filter_dataset_variants.py:51
Method
train_val_filter
(self, train_metadata, val_metadata)
robonet/datasets/variants/val_filter_dataset_variants.py:67
Function
traj_page
(traj_id)
scripts/visualize_dataset.py:30
Function
transpose_batch_time
(x)
robonet/video_prediction/utils/tf_utils.py:123
Function
trial_str_creator
(trial)
scripts/train_model.py:10
Function
upsample2d
(inputs, strides, padding='SAME', upsample_mode='bilinear')
robonet/video_prediction/ops.py:601
Function
upsample_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
Function
upsample_conv2d_v2
(inputs, filters, kernel_size, strides=(1, 1), padding='SAME', kernel=None, use_bias=Tr
robonet/video_prediction/ops.py:720
Method
vars
(self)
robonet/video_prediction/models/graphs/deterministic_graph.py:36
Method
vars
(self)
robonet/video_prediction/models/graphs/base_graph.py:17
Function
vgg16
rgb_image: 4-D tensor with pixel intensities between 0 and 1.
robonet/video_prediction/layers/vgg_network.py:38
Function
vgg19
rgb_image: 4-D tensor with pixel intensities between 0 and 1.
robonet/video_prediction/layers/vgg_network.py:105
Function
vgg_assign_from_values_fn
(model='vgg16', var_name_prefix='vgg/', var_name_k
robonet/video_prediction/layers/vgg_network.py:5
Function
vgg_cosine_distance
(image0, image1, keep_axis=None)
robonet/video_prediction/metrics.py:291
Function
vgg_cosine_distance_np
(image0, image1, keep_axis=None)
robonet/video_prediction/metrics.py:358
Method
zero_state
(self, batch_size, dtype)
robonet/video_prediction/layers/dnaflow_rnn_cell.py:257
← previous
301–384 of 384, ranked by callers