選択できるのは25トピックまでです。 トピックは、先頭が英数字で、英数字とダッシュ('-')を使用した35文字以内のものにしてください。

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556
  1. """ Wokers definition """
  2. # TODO Implement this with a queue and mutliprocessingt
  3. import threading
  4. from config import Config as Conf
  5. from transform.gan.mask import CorrectToMask, MaskrefToMaskdet, MaskfinToNude
  6. from transform.opencv.correct import DressToCorrect, ColorTransfer
  7. from transform.opencv.mask import MaskToMaskref, MaskdetToMaskfin
  8. from transform.opencv.resize import ImageToResized, ImageToCrop, ImageToOverlay, ImageToResizedCrop, ImageToRescale
  9. workers = {
  10. "gan": {
  11. CorrectToMask: [],
  12. MaskrefToMaskdet: [],
  13. MaskfinToNude: [],
  14. "sem": threading.Semaphore(1)
  15. },
  16. "opencv": {
  17. DressToCorrect: [],
  18. MaskToMaskref: [],
  19. ImageToResized: [],
  20. ImageToCrop: [],
  21. ImageToOverlay: [],
  22. ImageToResizedCrop: [],
  23. ImageToRescale: [],
  24. ColorTransfer: [],
  25. MaskdetToMaskfin: [],
  26. "sem": threading.Semaphore(Conf.args['n_cores'])
  27. }
  28. }
  29. select_sem = threading.Semaphore(1)
  30. def run_worker(klass, image_step, config=None):
  31. r = None
  32. for k in ("gan", "opencv"):
  33. if workers.get(k).get(klass) is not None:
  34. Conf.log.debug("wk {}".format(workers.get(k).get(klass)))
  35. workers.get(k).get("sem").acquire()
  36. select_sem.acquire()
  37. if len(workers.get(k).get(klass)) == 0:
  38. w = klass()
  39. else:
  40. w = workers.get(k).get(klass).pop(0)
  41. select_sem.release()
  42. r = w.run(*[image_step[i] for i in w.input_index], config=config)
  43. select_sem.acquire()
  44. workers.get(k).get(klass).append(w)
  45. select_sem.release()
  46. workers.get(k).get("sem").release()
  47. return r