Add script to test the simulation
This commit is contained in:
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()
|
Reference in New Issue
Block a user