MCPcopy Create free account

hub / github.com/Abraham190137/TactileACT / functions

Functions267 in github.com/Abraham190137/TactileACT

⨍ Functions 267◇ Types & classes 48
↓ 205 callersFunctionprint
(*args, **kwargs)
detr/detr/util/misc.py:370
↓ 103 callersMethodto
(self, device)
detr/detr/util/misc.py:289
↓ 14 callersMethodmax
(self)
detr/detr/util/misc.py:73
↓ 8 callersFunctionmodified_resnet18
Get a resnet18 model with all BatchNorm layers replaced with GroupNorm. weights: The weights to load into the model. If None, uses default pr
clip_pretraining.py:96
↓ 7 callersFunctionget_norm_stats
(dataset_dir, num_episodes, use_existing=True, chunk_size = 0)
diffusion/utils.py:187
↓ 6 callersFunctionget_norm_stats
(dataset_dir, num_episodes, use_existing=True, chunk_size = 0)
utils.py:208
↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
detr/detr/models/transformer.py:275
↓ 5 callersMethod__init__
(self, dim)
diffusion/network.py:56
↓ 5 callersMethodinput
(self, command)
robot_operation.py:215
↓ 4 callersMethod__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048
detr/detr/models/transformer.py:23
↓ 4 callersFunctionclip_loss
(image_embeddings:torch.Tensor, gelsight_embeddings:torch.Tensor, target_matrix:torch.Tensor, logit_scale = 1.
clip_pretraining.py:359
↓ 4 callersFunctionclip_loss
(image_embeddings:torch.Tensor, gelsight_embeddings:torch.Tensor, target_matrix:torch.Tensor, logit_scale = 1.
clip_pretraining_no_pos.py:350
↓ 4 callersFunctionclip_loss
(image_embeddings:torch.Tensor, gelsight_embeddings:torch.Tensor, target_matrix:torch.Tensor, logit_scale = 1.
failed_DDP.py:366
↓ 4 callersFunctionclip_loss
(image_embeddings, gelsight_embeddings, target_matrix, logit_scale = 1.0, visualize = False)
diffusion/clip_pretraining.py:227
↓ 4 callersFunctionload_ACT
Load the ACT model from the checkpoint file. model_path: str, path to the model checkpoint file. args_file: str, path to the args.json fi
load_ACT.py:7
↓ 4 callersFunctionmodified_resnet18
(features_per_group=16)
diffusion/clip_pretraining.py:92
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
detr/detr/models/backbone.py:90
↓ 3 callersFunction_run
(command)
detr/detr/util/misc.py:252
↓ 3 callersFunctionis_dist_avail_and_initialized
()
detr/detr/util/misc.py:378
↓ 3 callersFunctionmake_masks
(image_size, verticies)
diffusion/process_data_cage.py:25
↓ 3 callersMethodunnormalize
(self, qpos, action)
diffusion/utils.py:26
↓ 3 callersFunctionvisualize_gelsight_data
(image)
diffusion/robot_operation.py:57
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
detr/detr/models/transformer.py:396
↓ 2 callersFunction_get_clones
(module, N)
detr/detr/models/transformer.py:379
↓ 2 callersFunction_save_ckpt
(start_time:datetime,epoch,enc_type, nets,train_losses,val_losses,test=False)
diffusion/train_policy.py:148
↓ 2 callersFunctioncompute_dict_mean
(epoch_dicts)
utils.py:276
↓ 2 callersFunctioncreate_nets
(enc_type,data_dir,norm_stats,camera_names,pred_horizon, num_episodes=100)
diffusion/train_policy.py:37
↓ 2 callersFunctiondiffuse_robot
(qpos_data,image_data,camera_names,model_dict, pred_horizon,device=device)
diffusion/predict_robot_actions.py:13
↓ 2 callersFunctiongelsight_norm_stats
(dataset_dir, num_episodes)
utils.py:185
↓ 2 callersFunctiongelsight_norm_stats
(dataset_dir, num_episodes)
diffusion/utils.py:164
↓ 2 callersFunctionget_args_parser
()
detr/detr/main.py:12
↓ 2 callersFunctionget_resnet
name: resnet18, resnet34, resnet50 weights: "IMAGENET1K_V1", None
diffusion/network.py:281
↓ 2 callersFunctionget_world_size
()
detr/detr/util/misc.py:386
↓ 2 callersFunctionis_main_process
()
detr/detr/util/misc.py:398
↓ 2 callersFunctionmain
(args)
imitate_episodes.py:33
↓ 2 callersFunctionmake_masks
(image_size, verticies)
process_data_cage.py:25
↓ 2 callersFunctionmodified_resnet18
(weights=None, features_per_group=16)
clip_pretraining_no_pos.py:84
↓ 2 callersFunctionmodified_resnet18
(weights=None, features_per_group=16)
failed_DDP.py:95
↓ 2 callersFunctionplot_history
(train_history, validation_history, num_epochs, ckpt_dir, seed)
imitate_episodes.py:335
↓ 2 callersFunctionplot_run_similarity
Plots intra-run similarity matrix of the latent vectors all_vectors: np.ndarray of shape (episode_len, 512). The latent vectors.
clip_tsne.py:56
↓ 2 callersFunctionpredict_diff_actions
(batch,unnormalizer:NormalizeDiffusionActionQpos,model_dict,camera_names, device)
diffusion/visualize_waypts.py:52
↓ 2 callersMethodprocess_data
(self, images: Dict[str, np.ndarray], gelsight: np.ndarray,
diffusion/robot_operation.py:83
↓ 2 callersFunctionreplace_bn_with_gn
Relace all BatchNorm layers with GroupNorm.
diffusion/network.py:333
↓ 2 callersMethodunnormalize
(self, qpos, action)
utils.py:38
↓ 2 callersMethodunnormalize_delta
(self, normalized_delta)
utils.py:65
↓ 2 callersMethodupdate
(self, value, n=1)
detr/detr/util/misc.py:40
↓ 2 callersFunctionvisualize
(images, qpos, actions, ground_truth=None)
diffusion/robot_operation.py:128
↓ 2 callersFunctionvisualize
images = HxWxC qpos = 4, actions = nx4 ground_truth = nx4
diffusion/visualization.py:40
↓ 2 callersFunctionvisualize_gelsight_data
(image)
robot_operation.py:63
↓ 2 callersFunctionvisualize_gelsight_data
(image)
inspect_hdf5_file.py:11
↓ 2 callersFunctionvisualize_gelsight_data
(image)
diffusion/robot_operation_old.py:23
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
detr/detr/models/transformer.py:200
↓ 1 callersMethod__getitem__
(self, index)
utils.py:103
↓ 1 callersMethod__getitem__
(self, index)
diffusion/utils.py:82
↓ 1 callersMethod__init__
(self, episode_ids, dataset_dir, camera_names, norm_stats, chunk_size, image_size = None)
utils.py:76
↓ 1 callersMethod__init__
(self, episode_ids: List[int], dataset_dir: str, camera_
clip_pretraining.py:119
↓ 1 callersMethod__init__
(self, backbone, position_embedding)
policy.py:16
↓ 1 callersMethod__init__
(self, episode_ids: List[int], dataset_dir: str, camera_
clip_pretraining_no_pos.py:102
↓ 1 callersMethod__init__
(self, episode_ids: List[int], dataset_dir: str, camera_
failed_DDP.py:113
↓ 1 callersMethod__init__
(self, episode_ids, dataset_dir, camera_names, norm_stats, chunk_size, image_size = None)
diffusion/utils.py:55
↓ 1 callersMethod__init__
(self, episode_ids: List[int], dataset_dir: str, camera_
diffusion/clip_pretraining.py:110
↓ 1 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
detr/detr/models/position_encoding.py:19
↓ 1 callersFunction_max_by_axis
(the_list)
detr/detr/util/misc.py:275
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
detr/detr/util/misc.py:335
↓ 1 callersMethod_reset_parameters
(self)
detr/detr/models/transformer.py:45
↓ 1 callersFunctionbox_iou
(boxes1, boxes2)
detr/detr/util/box_ops.py:24
↓ 1 callersFunctionclip_pretraining
(train_loader: DataLoader, test_loader: DataLoader, device: torch.de
clip_pretraining.py:401
↓ 1 callersFunctionclip_pretraining
(train_loader: DataLoader, test_loader: DataLoader, device: torch.de
clip_pretraining_no_pos.py:392
↓ 1 callersFunctionclip_pretraining
(train_loader: DataLoader, test_loader: DataLoader, device: torch.de
failed_DDP.py:419
↓ 1 callersFunctiondetach_dict
(d)
utils.py:286
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
detr/detr/models/transformer.py:203
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
detr/detr/models/transformer.py:278
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
detr/detr/models/transformer.py:229
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
detr/detr/models/transformer.py:341
↓ 1 callersMethodgen_visualizations_dir
(self)
visualization_utils.py:31
↓ 1 callersMethodgen_visualizations_dir
(self)
diffusion/visualization.py:30
↓ 1 callersMethodget_gelsight
(self, episode_idx, timestep)
failed_DDP.py:310
↓ 1 callersMethodget_image
(self, episode_idx, timestep, cam_name)
failed_DDP.py:296
↓ 1 callersMethodget_latest_input
(self)
robot_operation.py:195
↓ 1 callersMethodget_position
(self, episode_idx, timestep)
failed_DDP.py:324
↓ 1 callersFunctionget_position_angle_vec
(position)
detrvae.py:18
↓ 1 callersFunctionget_rank
()
detr/detr/util/misc.py:392
↓ 1 callersFunctionget_sinusoid_encoding_table
(n_position, d_hid)
detrvae.py:17
↓ 1 callersFunctioninit_process
(rank, world_size, backend='nccl')
failed_DDP.py:12
↓ 1 callersFunctionkl_divergence
(mu:torch.Tensor, logvar:torch.Tensor)
policy.py:163
↓ 1 callersFunctionload_models
(dataset_dir, weights_dir, norm_stats, camera_names,num_episodes,pred_horizon)
diffusion/visualize_waypts.py:16
↓ 1 callersFunctionmonitor_cameras
(frames: Dict[str, np.ndarray], gelsight_frame: np.ndarray = None)
robot_operation.py:23
↓ 1 callersFunctionmonitor_cameras
(frames: Dict[str, np.ndarray], gelsight_frame: np.ndarray = None)
diffusion/robot_operation.py:17
↓ 1 callersFunctionnested_tensor_from_tensor_list
(tensor_list: List[Tensor])
detr/detr/util/misc.py:307
↓ 1 callersMethodnormalize_qpos
(self, qpos)
utils.py:26
↓ 1 callersFunctionplot_tsne
Plots the t-SNE visualization of the latent vectors of the images and gelsight data. all_image_vectors: np.ndarray of shape (n_cam, episode_l
clip_tsne.py:17
↓ 1 callersFunctionprint_hdf5_file
(filename)
inspect_hdf5_file.py:35
↓ 1 callersFunctionprint_hdf5_info
(group, indent="")
inspect_hdf5_file.py:21
↓ 1 callersMethodprocess_data
(self, images: Dict[str, np.ndarray], gelsight: np.ndarray,
robot_operation.py:90
↓ 1 callersMethodprocess_data
(self, images: Dict[str, np.ndarray], gelsight: np.ndarray,
diffusion/robot_operation_old.py:50
↓ 1 callersFunctionreparametrize
(mu, logvar)
detrvae.py:11
↓ 1 callersFunctionreplace_bn_with_gn
Relace all BatchNorm layers with GroupNorm.
clip_pretraining.py:50
↓ 1 callersFunctionreplace_bn_with_gn
Relace all BatchNorm layers with GroupNorm.
clip_pretraining_no_pos.py:44
↓ 1 callersFunctionreplace_bn_with_gn
Relace all BatchNorm layers with GroupNorm.
failed_DDP.py:55
↓ 1 callersFunctionreplace_bn_with_gn
Relace all BatchNorm layers with GroupNorm.
diffusion/clip_pretraining.py:58
next →1–100 of 267, ranked by callers