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

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