Реализация системы извлечения изображений по текстовому описанию и поиск похожих фотографий в датасете
Необходимо иметь установленный python 3 любой версии (лучше 11).
Данные команды требуется запускать последовательно в терминале:
- Склонируйте к себе этот репозиторий
git clone https://github.com/Koldim2001/image_retrieval.git
- Перейдите с помощью команды cd в созданную папку image_retrieval
cd SORT-DeepSORT-Tracker
- Загрузите все необходимые библиотеки:
pip install -r requirements.txt
PS: Лучше torch ставить сразу с поддержкой gpu если она имеется:
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
- Запустите ноутбук retrival.ipynb, в котором примеры кода по извлечению изображений
- Для запуска надо создать папку images в корне репозитория и положить туда желаемый набор фоток для анализа
Имеется подробный туториал по работе с данным репозиторием, в котором рассказаны основные теоретические и практические моменты по работе с image retrieval
Доступен по клику на иконку:
