main.py 376 B

123456789101112131415161718192021222324252627282930
  1. import sys
  2. import cv2
  3. from run import process
  4. """
  5. main.py
  6. How to run:
  7. python3 main.py
  8. """
  9. # ------------------------------------------------- main()
  10. def main():
  11. #Read input image
  12. dress = cv2.imread("input.png")
  13. #Process
  14. watermark = process(dress)
  15. # Write output image
  16. cv2.imwrite("output.png", watermark)
  17. #Exit
  18. sys.exit()
  19. if __name__ == '__main__':
  20. main()