Как повысить эффективность создания и хеширования изображений? - PullRequest
0 голосов
/ 27 мая 2020

Я использую код для создания случайного изображения, а затем проверяю значение ha sh этого изображения и сравниваю его с известным значением ha sh. Вот мой код:

import os
import time
import hashlib
import numpy
from io import BytesIO
from PIL import Image

def create_image(width = 1920, height = 1080, num_of_images = 100):
    width = int(width)
    height = int(height)
    num_of_images = int(num_of_images)

    current = time.strftime("%Y%m%d%H%M%S")
    os.mkdir(current)
    i=0
    for n in range(num_of_images):
        filename = '{0}/{0}_{1:03d}.jpg'.format(current, n)
        rgb_array = numpy.random.rand(height,width,3) * 255
        image = Image.fromarray(rgb_array.astype('uint8')).convert('RGB')
        image.save(filename)
        imagee = open(filename, "rb").read()
        hashnum = hashlib.md5(imagee).hexdigest()

        if(hashnum=="B3D740C2F83F7EE120FD16EAED266B43"):
            image.save(filename)
            print(filename)
        else:
            os.remove(filename)
        i+=1
        print("Done with ", i, end='\r')

def main(args):
    create_image(width = args[0], height = args[1], num_of_images = args[2])
    return 0

if __name__ == '__main__':
    import sys 
    status = main(sys.argv[1:])
    sys.exit(status)

Пока эта программа обрабатывает около 1k 17 изображений ширины 14 высот в секунду, и мне было интересно, может ли это быть более эффективным. Я думал о том, чтобы избавиться от всего сохраненного изображения, а затем удалить, но я не совсем уверен, как это сделать. Также извините, если в этом сообщении есть ошибки, английский sh не мой родной язык.

1 Ответ

0 голосов
/ 27 мая 2020

Этот ответ показывает вам, как избежать необходимости сохранять образ на диск перед его хешированием.

Удаление записи / чтения с диска должно сделать его значительно быстрее

...