measure_energy_iot/image_compression.py

   1 from PIL import Image
   2 import time
   3 import cv2
   4
   5 import sys
   6
   7 quality=int(sys.argv[1])
   8 print("quality ",quality)
   9
  10 im=Image.open("lena.jpg")
  11 elapsed_time=0
  12
  13 for i in range(100):
  14     t = time.process_time()
  15     im.save("lena2.jpg",quality=quality)
  16     elapsed_time += time.process_time() - t
  17
  18 print("time pil",elapsed_time)
  19
  20
  21
  22
  23
  24 elapsed_time=0
  25 for i in range(100):
  26     t = time.process_time()
  27     im2=im.resize((256, 256))
  28     im2.save("lena3.jpg",quality=quality)
  29     elapsed_time += time.process_time() - t
  30
  31 print("time resize + pil",elapsed_time)
  32
  33
  34
  35
  36 img = cv2.imread('lena.jpg')
  37 elapsed_time=0
  38
  39
  40 for i in range(100):
  41     t = time.process_time()
  42     cv2.imwrite("lena4.jpg", img,[int(cv2.IMWRITE_JPEG_QUALITY), quality])
  43     elapsed_time += time.process_time() - t
  44
  45 print("time cv2",elapsed_time)
  46
  47
  48
  49 for i in range(100):
  50     t = time.process_time()
  51     resized = cv2.resize(img, (256,256), interpolation=cv2.INTER_LINEAR)
  52     cv2.imwrite("lena5.jpg", resized,[int(cv2.IMWRITE_JPEG_QUALITY), quality])
  53     elapsed_time += time.process_time() - t
  54
  55 print("time resize + cv2",elapsed_time)