Есть ли способ извлечь размер изображения модели darknet в Python?

Я использую darknet в проекте Python (3.5) для обнаружения объектов в изображениях. Поскольку при тестировании я использую сеть разных размеров, мне очень удобно иметь возможность извлекать размер изображения, который требуется сети. Что-то вроде:

from darknet import Darknet

model = Darknet('my_config.cfg')
height, width = model.get_image_size() # Example

Я помню проект с открытым исходным кодом, извлекающий его из информации о слоях модели, но я не могу вспомнить, как они это сделали.

0 ответов

Другие вопросы по тегам