Você não pode selecionar mais de 25 tópicos Os tópicos devem começar com uma letra ou um número, podem incluir traços ('-') e podem ter até 35 caracteres.

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748
  1. """Multiple Image Transform Processing."""
  2. import copy
  3. from multiprocessing.pool import ThreadPool
  4. from config import Config as Conf
  5. from processing import Processing, SimpleProcessing
  6. from utils import camel_case_to_str
  7. class MultipleImageProcessing(Processing):
  8. """Multiple Image Processing Class."""
  9. def _setup(self, *args):
  10. self._input_paths = self._args['input']
  11. self._output_paths = self._args['output']
  12. self._process_list = []
  13. self._multiprocessing = Conf.multiprocessing()
  14. self._process_list = []
  15. for input_path, output_path in zip(self._input_paths, self._output_paths):
  16. args = copy.deepcopy(self._args)
  17. args['input'] = input_path
  18. args['output'] = output_path
  19. self._process_list.append((SimpleProcessing(args), args))
  20. Conf.log.debug(self._process_list)
  21. def _process_one(self, a):
  22. Conf.log.info("{} : {}/{}".format(
  23. camel_case_to_str(self.__class__.__name__), a[1] + 1, len(self._process_list)
  24. ))
  25. a[0][0].run(config=a[0][1])
  26. def _execute(self, *args):
  27. """
  28. Execute all phases on the list of images.
  29. :return: None
  30. """
  31. if not self._multiprocessing:
  32. for x in zip(self._process_list, range(len(self._process_list))):
  33. self._process_one(x)
  34. else:
  35. Conf.log.debug("Using Multiprocessing")
  36. pool = ThreadPool(Conf.args['n_cores'])
  37. pool.map(self._process_one, zip(self._process_list, range(len(self._process_list))))
  38. pool.close()
  39. pool.join()