deal_cutout.py 2.0 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061
  1. import time
  2. from concurrent.futures import as_completed, ThreadPoolExecutor, wait
  3. import threading
  4. from .remove_bg_pixian import RemoveBgPiXian
  5. from .other.module_online_data import GetOnlineData
  6. from .deal_one_image import DealOneImage, DealOneImageBeforehand
  7. from .other.log import MyLogger
  8. class DealCutout:
  9. def __init__(self):
  10. super().__init__()
  11. self.lock = threading.Lock()
  12. self.need_cutout_images = {}
  13. self.state = 2 # 1进行中 2停止
  14. self.get_online_data = GetOnlineData()
  15. self.is_upload_pic_num = 0
  16. self.is_deal_num = 0
  17. # 图片列表
  18. self.upload_pic_dict = {}
  19. self.logger = MyLogger().logger
  20. def startDispose(self):
  21. self.get_online_data.refresh_headers()
  22. num = 0
  23. for image_data in self.need_cutout_images:
  24. num += 1
  25. upload_pic_dict = {}
  26. upload_pic_dict = DealOneImageBeforehand(
  27. image_data=image_data, lock=self.lock, windows=self, num=num
  28. ).run(upload_pic_dict)
  29. print("upload_pic_dict", upload_pic_dict)
  30. DealOneImage(
  31. image_data=image_data, lock=self.lock, windows=self, num=num
  32. ).run(image_data, upload_pic_dict)
  33. class DealCloths:
  34. def __init__(self):
  35. super().__init__()
  36. self.lock = threading.Lock()
  37. self.need_cutout_images = {}
  38. self.state = 2 # 1进行中 2停止
  39. self.get_online_data = GetOnlineData()
  40. self.is_upload_pic_num = 0
  41. self.is_deal_num = 0
  42. # 图片列表
  43. self.upload_pic_dict = {}
  44. self.logger = MyLogger().logger
  45. def startDispose(self):
  46. self.get_online_data.refresh_headers()
  47. num = 0
  48. for image_data in self.need_cutout_images:
  49. num += 1
  50. upload_pic_dict = {}
  51. hand = DealOneImageBeforehand(
  52. image_data=image_data, lock=self.lock, windows=self, num=num
  53. )
  54. upload_pic_dict = hand.get_image_cut_cloths()