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