MCPcopy Create free account

hub / github.com/MarkFzp/act-plus-plus / functions

Functions274 in github.com/MarkFzp/act-plus-plus

⨍ Functions 274◇ Types & classes 40
↓ 120 callersFunctionprint
(*args, **kwargs)
detr/util/misc.py:370
↓ 11 callersMethodmax
(self)
detr/util/misc.py:73
↓ 8 callersMethodto
(self, device)
detr/util/misc.py:289
↓ 6 callersMethodread
Reads data from the motors.
dynamixel_client.py:374
↓ 6 callersFunctionset_seed
(seed)
utils.py:358
↓ 6 callersFunctionunsigned_to_signed
Converts the given value from its unsigned representation.
dynamixel_client.py:52
↓ 6 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
detr/models/transformer.py:226
↓ 5 callersMethodinterpolate
(curr_waypoint, next_waypoint, t)
scripted_policy.py:23
↓ 5 callersFunctionmake_sim_env
Environment for simulated robot bi-manual manipulation, with joint position control Action space: [left_arm_qpos (6), # abso
sim_env.py:20
↓ 4 callersMethod__init__
Initializes a new client. Args: motor_ids: All motor IDs being used by the client. port: The Dynamixel device to talk
dynamixel_client.py:69
↓ 4 callersMethod__init__
(self, d_model=512, nhead=8, num_encoder_layers=6, num_decoder_layers=6, dim_feedforward=2048
detr/models/transformer.py:22
↓ 4 callersFunctioncompute_dict_mean
(epoch_dicts)
utils.py:342
↓ 4 callersMethoddeserialize
(self, model_dict)
policy.py:251
↓ 4 callersFunctionflatten_list
(l)
utils.py:15
↓ 4 callersFunctionsample_box_pose
()
utils.py:304
↓ 4 callersMethodserialize
(self)
policy.py:248
↓ 3 callersMethod__init__
(self, backbone, position_embedding)
detr/models/backbone.py:100
↓ 3 callersFunction_run
(command)
detr/util/misc.py:252
↓ 3 callersMethodcheck_connected
Ensures the robot is connected.
dynamixel_client.py:299
↓ 3 callersMethodconfigure_optimizers
(self)
policy.py:236
↓ 3 callersMethoddisconnect
Disconnects from the Dynamixel device.
dynamixel_client.py:167
↓ 3 callersFunctionfind_all_hdf5
(dataset_dir, skip_mirrored_data)
utils.py:198
↓ 3 callersMethodhandle_packet_result
Handles the result from a communication request.
dynamixel_client.py:306
↓ 3 callersFunctionis_dist_avail_and_initialized
()
detr/util/misc.py:378
↓ 3 callersFunctionmake_ee_sim_env
Environment for simulated robot bi-manual manipulation, with end-effector control. Action space: [left_arm_pose (7), # posit
ee_sim_env.py:20
↓ 3 callersFunctionsave_videos
(video, dt, video_path=None)
visualize_episodes.py:48
↓ 2 callersFunctionBatchSampler
(batch_size, episode_len_l, sample_weights)
utils.py:209
↓ 2 callersMethod__init__
(self, random=None)
sim_env.py:173
↓ 2 callersMethod__init__
(self, args_override)
policy.py:200
↓ 2 callersMethod__init__
(self, prediction_len)
train_actuator_network.py:213
↓ 2 callersMethod__init__
(self, random=None)
ee_sim_env.py:202
↓ 2 callersFunction_get_activation_fn
Return an activation function given a string
detr/models/transformer.py:306
↓ 2 callersFunction_get_clones
(module, N)
detr/models/transformer.py:289
↓ 2 callersMethod_get_data
Returns a copy of the data.
dynamixel_client.py:414
↓ 2 callersFunctionbuild_backbone
(args)
detr/models/backbone.py:115
↓ 2 callersFunctionbuild_transformer
(args)
detr/models/transformer.py:293
↓ 2 callersFunctionchunks
Yield successive n-sized chunks from lst.
vinn_select_k.py:37
↓ 2 callersMethodconnect
Connects to the Dynamixel motors. NOTE: This should be called after all DynamixelClients on the same process are created.
dynamixel_client.py:142
↓ 2 callersMethodencode
(self, qpos, actions=None, is_pad=None, vq_sample=None)
detr/models/detr_vae.py:90
↓ 2 callersFunctioneval_bc
(config, ckpt_name, save_episode=True, num_rollouts=50)
imitate_episodes.py:228
↓ 2 callersFunctionforward_pass
(data, policy)
imitate_episodes.py:529
↓ 2 callersFunctionforward_pass
(data, policy, latent_model)
train_latent_model.py:323
↓ 2 callersMethodgenerate
(self, n, temperature=0.1, x=None)
detr/models/latent_model.py:62
↓ 2 callersFunctionget_args_parser
()
detr/main.py:12
↓ 2 callersFunctionget_norm_stats
(dataset_path_list)
utils.py:149
↓ 2 callersFunctionget_world_size
()
detr/util/misc.py:386
↓ 2 callersMethodinitialize_episode
Sets the state of the environment at the start of each episode.
sim_env.py:177
↓ 2 callersMethodinitialize_episode
Sets the state of the environment at the start of each episode.
ee_sim_env.py:206
↓ 2 callersMethodinitialize_robots
(self, physics)
ee_sim_env.py:76
↓ 2 callersFunctionis_main_process
()
detr/util/misc.py:398
↓ 2 callersFunctionload_data
(dataset_dir_l, name_filter, camera_names, batch_size_train, batch_size_val, chunk_size, skip_mirrored_data=Fa
utils.py:220
↓ 2 callersFunctionmake_policy
(policy_class, policy_config)
imitate_episodes.py:182
↓ 2 callersFunctionplot_history
(train_history, validation_history, num_epochs, ckpt_dir, seed)
train_latent_model.py:431
↓ 2 callersFunctionpreprocess_base_action
(base_action)
utils.py:287
↓ 2 callersMethodread_pos_vel_cur
Returns the current positions and velocities.
dynamixel_client.py:210
↓ 2 callersFunctionrepeater
(data_loader)
imitate_episodes.py:624
↓ 2 callersFunctionsample_insertion_pose
()
utils.py:315
↓ 2 callersFunctionsave_videos
(video, dt, video_path=None)
truncate_data.py:65
↓ 2 callersMethodupdate
(self, value, n=1)
detr/util/misc.py:40
↓ 2 callersFunctionvisualize_prediction
(dataset_path_list, episode_ids, policy, norm_stats, history_len, future_len, prediction_len, ckpt_dir, step,
train_actuator_network.py:150
↓ 2 callersMethodwith_pos_embed
(self, tensor, pos: Optional[Tensor])
detr/models/transformer.py:163
↓ 1 callersMethod__getitem__
(self, index)
utils.py:48
↓ 1 callersMethod__getitem__
(self, index)
train_actuator_network.py:334
↓ 1 callersMethod__init__
Initializes the model. Parameters: backbones: torch module of the backbone to be used. See backbone.py transformer: t
detr/models/detr_vae.py:184
↓ 1 callersMethod__init__
(self, num_pos_feats=64, temperature=10000, normalize=False, scale=None)
detr/models/position_encoding.py:19
↓ 1 callersMethod__init__
(self, seq_len, latent_dim, num_head)
detr/models/latent_model.py:9
↓ 1 callersMethod_initialize_data
Initializes the cached data.
dynamixel_client.py:405
↓ 1 callersMethod_locate_transition
(self, index)
utils.py:41
↓ 1 callersMethod_locate_transition
(self, index)
train_actuator_network.py:327
↓ 1 callersFunction_max_by_axis
(the_list)
detr/util/misc.py:275
↓ 1 callersFunction_onnx_nested_tensor_from_tensor_list
(tensor_list: List[Tensor])
detr/util/misc.py:335
↓ 1 callersMethod_reset_parameters
(self)
detr/models/transformer.py:44
↓ 1 callersMethod_update_data
Updates the data index for the given motor ID.
dynamixel_client.py:409
↓ 1 callersFunctionbox_iou
(boxes1, boxes2)
detr/util/box_ops.py:24
↓ 1 callersFunctionbuild_ACT_model
(args)
detr/models/__init__.py:5
↓ 1 callersFunctionbuild_ACT_model_and_optimizer
(args_override)
detr/main.py:86
↓ 1 callersFunctionbuild_CNNMLP_model
(args)
detr/models/__init__.py:8
↓ 1 callersFunctionbuild_CNNMLP_model_and_optimizer
(args_override)
detr/main.py:109
↓ 1 callersFunctionbuild_cnnmlp
(args)
detr/models/detr_vae.py:305
↓ 1 callersFunctionbuild_position_encoding
(args)
detr/models/position_encoding.py:83
↓ 1 callersFunctioncalculate_nearest_neighbors
(query_inputs, query_targets, support_inputs, support_targets, max_k)
vinn_select_k.py:15
↓ 1 callersFunctioncalculate_nearest_neighbors
(curr_feature, support_inputs, support_targets, k, state_weight)
vinn_eval.py:28
↓ 1 callersFunctionchunks
Yield successive n-sized chunks from lst.
vinn_cache_feature.py:20
↓ 1 callersFunctioncompress_dataset
(input_dataset_path, output_dataset_path)
truncate_data.py:19
↓ 1 callersFunctioncompress_dataset
(input_dataset_path, output_dataset_path)
compress_data.py:18
↓ 1 callersFunctiondetach_dict
(d)
utils.py:352
↓ 1 callersMethodforward
(self, x)
detr/models/latent_model.py:53
↓ 1 callersMethodforward_post
(self, src, src_mask: Optional[Tensor] = None,
detr/models/transformer.py:166
↓ 1 callersMethodforward_post
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask:
detr/models/transformer.py:229
↓ 1 callersMethodforward_pre
(self, src, src_mask: Optional[Tensor] = None, src_key_padding_mask: O
detr/models/transformer.py:181
↓ 1 callersMethodforward_pre
(self, tgt, memory, tgt_mask: Optional[Tensor] = None, memory_mask: Op
detr/models/transformer.py:252
↓ 1 callersMethodgenerate_trajectory
(self, ts_first)
scripted_policy.py:19
↓ 1 callersFunctionget_action
(master_bot_left, master_bot_right)
sim_env.py:234
↓ 1 callersFunctionget_auto_index
(dataset_dir)
imitate_episodes.py:30
↓ 1 callersMethodget_env_state
(physics)
sim_env.py:101
↓ 1 callersMethodget_env_state
(physics)
ee_sim_env.py:128
↓ 1 callersFunctionget_image
(ts, camera_names)
vinn_eval.py:312
↓ 1 callersFunctionget_image
(ts, camera_names, rand_crop_resize=False)
imitate_episodes.py:206
↓ 1 callersFunctionget_norm_stats
(dataset_path_list)
train_actuator_network.py:271
↓ 1 callersFunctionget_position_angle_vec
(position)
detr/models/detr_vae.py:25
next →1–100 of 274, ranked by callers