main.py 915 B

123456789101112131415161718192021
  1. from Simsim import Simsim
  2. import numpy as np
  3. import logging
  4. if __name__ == "__main__":
  5. s = Simsim()
  6. s.rate = 5
  7. s.add_tracker('tracker', np.array([500, 300]), sensor_rad=250)
  8. s.add_tracker('tracker', np.array([300, 500]), sensor_rad=250)
  9. s.add_tracker('tracker', np.array([700, 500]), sensor_rad=250)
  10. # s.add_tracker('tracker', np.array([700, 580]), sensor_rad=200)
  11. # s.add_tracker('tracker', np.array([620, 220]), sensor_rad=450)
  12. # s.add_target('tgt', np.array([150, 50]))
  13. s.add_target('tgt', np.array([500, 490]))
  14. # s.add_target('tgt', np.array([180, 500]))
  15. s.add_target('tgt', np.array([500, 300]))
  16. s.add_target('tgt', np.array([300, 500]))
  17. # s.add_target('tgt', np.array([630, 500]))
  18. # s.add_edges([[0, 1], [0, 2], [3, 2], [1, 3], [1, 4], [3, 4]])
  19. s.add_edges([[0, 1], [1, 2], [0, 2]])
  20. s.run(logging.DEBUG, ground_truth=True)