Add script to test the simulation
This commit is contained in:
parent
0cdc5dd127
commit
1dd6f7875e
@ -1,12 +0,0 @@
|
||||
import rospy
|
||||
|
||||
from active_grasp.srv import *
|
||||
|
||||
rospy.init_node("reset")
|
||||
|
||||
seed = rospy.ServiceProxy("seed", Seed)
|
||||
reset = rospy.ServiceProxy("reset", Reset)
|
||||
|
||||
seed(SeedRequest(1))
|
||||
reset(ResetRequest())
|
||||
rospy.sleep(1.0)
|
13
test/test_sim_scene.py
Normal file
13
test/test_sim_scene.py
Normal file
@ -0,0 +1,13 @@
|
||||
from active_grasp.simulation import Simulation
|
||||
|
||||
|
||||
def main():
|
||||
gui = True
|
||||
scene_id = "random"
|
||||
vgn_path = "../vgn/assets/models/vgn_conv.pth"
|
||||
sim = Simulation(gui, scene_id, vgn_path)
|
||||
bbox = sim.reset()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
Loading…
x
Reference in New Issue
Block a user