Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/Vinson-sheep/multi_rotor_avoidance_rl
/ functions
Functions
100 in github.com/Vinson-sheep/multi_rotor_avoidance_rl
⨍
Functions
100
◇
Types & classes
18
↓ 12 callers
Method
load
(self)
test/embedding/embedding_learner.py:115
↓ 12 callers
Method
load
(self)
scripts/SAC.py:256
↓ 11 callers
Method
step
game step
scripts/common/game.py:398
↓ 9 callers
Method
save
(self)
scripts/SAC.py:247
↓ 9 callers
Method
size
size of buffer
test/embedding/embedding_buffer.py:55
↓ 6 callers
Method
sample
sample from buffer according to priority
scripts/common/buffer.py:66
↓ 5 callers
Method
_send_position_cmd
send position command on local ENU
scripts/common/game.py:654
↓ 5 callers
Method
sample_available
True if len(self.memory) >= self.batch_size
scripts/common/buffer.py:99
↓ 4 callers
Method
_is_hold
(self)
scripts/common/game.py:583
↓ 4 callers
Method
add
arg: - transition: the payload needed to be stored - prority: the priority of transition
scripts/common/buffer.py:33
↓ 4 callers
Method
is_crashed
determine if the uav is crashed. return: - True if it was crashed. - crash reward
scripts/common/game.py:333
↓ 4 callers
Method
size
size of buffer
scripts/common/buffer.py:106
↓ 4 callers
Method
start
send take-off command
scripts/common/game.py:241
↓ 4 callers
Method
update_priorities
update priorities.
scripts/common/buffer.py:88
↓ 3 callers
Method
_send_velocity_cmd
send velocity command on body end
scripts/common/game.py:630
↓ 3 callers
Method
cur_state
35 laser 1 vx 1 yaw_rate 1 ax 1 ayaw 1 distance 1 angle_dif
scripts/common/game.py:677
↓ 3 callers
Method
reset
reset uav and target
scripts/common/game.py:170
↓ 3 callers
Method
set_target
reset target ball
scripts/common/world.py:191
↓ 2 callers
Method
Q1
(self, state, action)
scripts/TD3.py:93
↓ 2 callers
Method
__init__
(self, **kwargs)
scripts/DDPG.py:81
↓ 2 callers
Method
__init__
(self, **kwargs)
scripts/SAC.py:126
↓ 2 callers
Method
__init__
(self, **kwargs)
scripts/TD3.py:105
↓ 2 callers
Method
act
(self, state)
scripts/SAC.py:161
↓ 2 callers
Method
add
add data to buffer
test/embedding/embedding_buffer.py:24
↓ 2 callers
Method
clear
clear world
scripts/common/world.py:166
↓ 2 callers
Method
evaluate
(self, state)
scripts/SAC.py:73
↓ 2 callers
Method
sample
sample from buffer according to priority
test/embedding/embedding_buffer.py:37
↓ 2 callers
Method
save
(self)
test/embedding/embedding_learner.py:109
↓ 1 callers
Method
Q1
(self, s, a)
scripts/SAC.py:114
↓ 1 callers
Method
__init__
(self, **kwargs)
test/embedding/embedding_learner.py:61
↓ 1 callers
Method
__init__
(self)
scripts/training_node.py:79
↓ 1 callers
Method
_is_arrived
(self, x, y, z, limit=0.1)
scripts/common/game.py:592
↓ 1 callers
Method
check_safe
check whether (x, y) falls into safe space. return: True - if not fall into
scripts/common/world.py:246
↓ 1 callers
Method
forward
(self, x)
scripts/SAC.py:64
↓ 1 callers
Method
init_wall_param
(self, safe_space, safe_radius)
scripts/common/world.py:219
↓ 1 callers
Method
is_arrived
(self)
scripts/common/game.py:597
↓ 1 callers
Method
last_loss
(self)
test/embedding/embedding_learner.py:106
↓ 1 callers
Method
learn
(self)
test/embedding/embedding_learner.py:81
↓ 1 callers
Function
load
()
scripts/testing_node.py:76
↓ 1 callers
Method
load
load data
test/embedding/embedding_buffer.py:76
↓ 1 callers
Method
load
(self)
scripts/DDPG.py:181
↓ 1 callers
Method
load
(self)
scripts/TD3.py:221
↓ 1 callers
Method
load
load data
scripts/common/buffer.py:131
↓ 1 callers
Function
loadData
()
scripts/training_node.py:136
↓ 1 callers
Method
put
(self, *transition)
scripts/SAC.py:170
↓ 1 callers
Method
recovery
run recovery action (keep away from obstacles in slight velocity) return: - True if crash when recovering.
scripts/common/game.py:505
↓ 1 callers
Method
reset
reset world
scripts/common/world.py:69
↓ 1 callers
Function
save
(episode, success_num, crash_num)
scripts/testing_node.py:68
↓ 1 callers
Method
save
save data
test/embedding/embedding_buffer.py:62
↓ 1 callers
Method
save
save data
scripts/common/buffer.py:113
↓ 1 callers
Method
update
(self)
scripts/SAC.py:179
Method
__init__
(self, input_dim, output_dim)
test/embedding/embedding_learner.py:43
Method
__init__
Embedding buffer is a container storing state data. The implementation here is for embedding layer constructing.
test/embedding/embedding_buffer.py:13
Method
__init__
(self, state_dim, action_dim, init_w=3e-3)
scripts/DDPG.py:41
Method
__init__
(self, state_dim, action_dim, init_w=3e-3)
scripts/DDPG.py:61
Method
__init__
(self)
scripts/training_node.py:107
Method
__init__
(self, state_dim, action_dim, init_w=3e-3)
scripts/SAC.py:47
Method
__init__
(self, state_dim, action_dim, init_w=3e-3)
scripts/SAC.py:85
Method
__init__
(self, state_dim, action_dim, init_w=3e-3)
scripts/TD3.py:44
Method
__init__
(self, state_dim, action_dim, init_w=3e-3)
scripts/TD3.py:64
Method
__init__
(self)
scripts/common/laser_filter.py:12
Method
__init__
safe_space: safe_radius: wall_rate: thr propotion of having wall. cylidar_num:
scripts/common/world.py:14
Method
__init__
initialize
scripts/common/game.py:26
Method
__init__
prioritized replay buffer implemented by Pytorch
scripts/common/buffer.py:15
Method
_bodyVelocityCB
(self, msg)
scripts/common/game.py:534
Method
_hold
if no cmd, hold command is sent.
scripts/common/game.py:547
Method
_mavrosStateCB
(self, msg)
scripts/common/game.py:528
Method
_scanCB
(self, msg)
scripts/common/game.py:531
Method
_vision
get iris state, and send vision.
scripts/common/game.py:605
Function
accCB
(msg)
test/reboot.py:19
Method
act
(self, state)
scripts/DDPG.py:105
Method
act
(self, state)
scripts/TD3.py:129
Method
alpha
(self)
scripts/SAC.py:158
Function
cmdCB
(msg)
test/plot.py:42
Function
cmdCB
(msg)
test/var_collect.py:37
Method
forward
(self, state)
test/embedding/embedding_learner.py:50
Method
forward
(self, state)
scripts/DDPG.py:51
Method
forward
(self, state, action)
scripts/DDPG.py:70
Method
forward
(self, s, a)
scripts/SAC.py:102
Method
forward
(self, state)
scripts/TD3.py:54
Method
forward
(self, state, action)
scripts/TD3.py:81
Method
is_valid
determine whether action in state after time_step is valid return: - True if action is valid - valid reward
scripts/common/game.py:357
Method
laserCB
(self, msg)
scripts/common/laser_filter.py:18
Method
put
(self, *transition)
scripts/DDPG.py:109
Method
put
(self, *transition)
scripts/TD3.py:133
Function
rawCB
(msg)
test/plot.py:65
Function
realCB
(msg)
test/var_collect.py:76
Function
resetCB
(event)
test/var_collect.py:80
Method
run
(self)
scripts/training_node.py:82
Method
run
(self)
scripts/training_node.py:110
Method
save
(self)
scripts/DDPG.py:174
Method
save
(self)
scripts/TD3.py:213
Function
scanCB
(msg)
test/reboot.py:15
Method
set_cylinder_rate
(self, cylinder_rate)
scripts/common/world.py:216
Method
set_wall_rate
(self, wall_rate)
scripts/common/world.py:213
Function
stateCB
(msg)
test/embedding/embedding_data_collect.py:21
Method
state_mean_std
(self)
scripts/common/buffer.py:63
Function
twistCB
(msg)
test/plot.py:61
Method
update
(self)
scripts/DDPG.py:118
Method
update
(self)
scripts/TD3.py:142