оцените объем используемой памяти при загрузке набора изображений - PullRequest
0 голосов
/ 07 марта 2019

Я работаю над проблемой классификации кошек и собак.Мой ПК на данный момент имеет только 8 ГБ ОЗУ, поэтому при использовании некоторых из них у меня остается около 6 ГБ.

Началось с попытки загрузить образ test_set, который содержит 12 500 изображений и занимает 285 МБ на жестком диске.Запуская скрипт, он использует всю доступную оперативную память и, скорее всего, просто остановит компьютер, поэтому я остановил его до того, как он превысит 95% используемого.

import os

test_images = os.listdir('data/test')
print(len(test_images))
>>> 12500

test_images_data = []

for image in test_images:
    image_data = cv2.imread('data/test/' + image)

    #convert color from BGR to RGB
    image_data = cv2.cvtColor(image_data, cv2.COLOR_BGR2RGB)

    test_images.append(image_data)

Есть ли способ узнать, сколько это оперативной памятискорее всего, использовать заранее, поэтому я знаю, например, будет ли достаточно добавить 8 ГБ ОЗУ?

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...