open reid

Открытый

Open-Reid-это легкая библиотека личностного лица для повторной идентификации для исследований. Он направлен на предоставление единого интерфейса для различных наборов данных, полный набор моделей и показателей оценки, а также примеры для воспроизведения (почти) самых современных результатов.

Установка

Установите Pytorch (версия> = 0.2.0). Хотя мы поддерживаем как Python2, так и Python3, мы рекомендуем Python3 для лучшей производительности.

git clone https://github.com/Cysu/open-reid.git
cd open-reid
python setup.py install

Примеры

python examples/softmax_loss.py -d viper -b 64 -j 2 -a resnet50 --logs-dir logs/softmax-loss/viper-resnet50

Это просто быстрый пример. Набор данных Viper может быть недостаточно большим, чтобы тренировать глубокую нейронную сеть.

Проверьте больше примеров и тестов.