Add script to test the simulation

This commit is contained in:
Michel Breyer 2021-11-09 11:48:18 +01:00
parent 0cdc5dd127
commit 1dd6f7875e
2 changed files with 13 additions and 12 deletions

View File

@ -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
View 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()