https://github.com/sadevans/openeyesclassifier
Тестовое для VisionLabs на CV engineer
https://github.com/sadevans/openeyesclassifier
Last synced: 3 months ago
JSON representation
Тестовое для VisionLabs на CV engineer
- Host: GitHub
- URL: https://github.com/sadevans/openeyesclassifier
- Owner: sadevans
- Created: 2024-08-07T07:18:16.000Z (10 months ago)
- Default Branch: main
- Last Pushed: 2024-08-13T16:31:29.000Z (10 months ago)
- Last Synced: 2025-01-19T00:46:21.531Z (5 months ago)
- Language: Python
- Size: 3.09 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# OpenEyesClassificator
Тестовое для VisionLabs на CV engineer
Отчет о проделанной работе: https://beautiful-fang-60a.notion.site/1eebfbe5a4d745cbb453dc318c373fac
Создайте виртуальную среду:
```bash
python -m venv venv
```
Активируйте ее:
```bash
source venv/bin/activate
```
Установите необходимые для инференса библиотеки:
```bash
pip install -r requirements.txt
```
Чтобы использовать классифкатор импортируйте из файла OpenEyesClassificator класс OpenEyesClassificator. Также убедитесь, что рядом с файлом `OpenEyesClassificator.py` лежат веса модели ``open_eyes_classifier.pth``.
```python
from OpenEyesClassificator import OpenEyesClassificator
```
Затем вы можете его иницилизовать и использовать его метод predict
```
model = OpenEyesClassificator()
model.predict(inpIm)
```
## Для воспроизводимости
Распакуйте папку dataset.zip, которая находится в директории `dataset`. Также создайте директории для сохранения экспериментов.