check_renderer.py 1.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142
  1. import os
  2. import argparse
  3. from rawls.rawls import Rawls
  4. from rawls.stats import RawlsStats
  5. def main():
  6. parser = argparse.ArgumentParser(description="concat and mean data and compare with 1 image only")
  7. parser.add_argument('--folder', type=str, help="folder which contains rawls files")
  8. parser.add_argument('--n', type=int, help="number of images to concat and mean")
  9. parser.add_argument('--output', type=str, help="output data folder")
  10. args = parser.parse_args()
  11. p_folder = args.folder
  12. p_n = args.n
  13. p_output = args.output
  14. files = os.listdir(p_folder)
  15. paths = [ os.path.join(p_folder, f) for f in files ]
  16. if not os.path.exists(p_output):
  17. os.makedirs(p_output)
  18. print('Loading 1 rawls file')
  19. rawls_one = Rawls.load(paths[0])
  20. print('Saving 1 rawls file into png')
  21. rawls_one.save(os.path.join(p_output, 'rawls_one.png' ))
  22. print('Loading {0} rawls files'.format(p_n))
  23. rawls_n = RawlsStats.load(paths[0:p_n])
  24. print('Saving {0} rawls files into png'.format(p_n))
  25. rawls_n.mean().save(os.path.join(p_output, 'rawls_{0}.png'.format(p_n)))
  26. if __name__ == "__main__":
  27. main()