nbv_sim/test/test_sim_scene.py

14 lines
260 B
Python
Raw Normal View History

2021-11-09 11:48:18 +01:00
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()