CVCounter - это приложение для подсчета объектов с использованием компьютерного зрения, реализованное на Python с использованием Flask. Проект предоставляет три режима отображения данных: основной вид с видео, текстовый вид и текстовый вид с двумя счетчиками.
- Клонируйте репозиторий:
git clone https://github.com/BespredeL/CVCounter.git
- Перейдите в директорию проекта:
cd CVCounter - Установите virtual environment:
python3 -m venv venv
- Активируйте virtual environment:
source venv/bin/activate - Установите зависимости:
pip3 install -r requirements.txt
- Переименуйте файл конфигурации:
mv config.example.json config.json
- Измените параметры в файле конфигурации, добавьте свою модель YOLO.
- Запустите приложение:
python app.py
В этом решение реализовано 3 вида просмотра:
- Основной вид - страница на которой выводится значение счетчиков и видео с результатом распознавания
- Текстовый вид - страница на которой выводится только значение счетчиков
- Текстовый вид с двумя счетчиками - страница на которой выводится значение 2 счетчиков (например на входе и выходе)
После нескольких вариантов, принял решение реализации на Flask, т.е решение в виде мини веб-сайта, так как такое решение позволяет избежать установки какого либо дополнительного софта на клиенты. А так же это решение не требовательно к клиентам в плане потребления ресурсов (за исключением основного вида с видео)
Мне удалось запустить 6 одновременных подсчетов (без вывода видео), и 5 подсчета с выводом видео.
Характеристики сервера:
- AMD Ryzen 5 3600
- GeForce GTX 1050 Ti (4Гб)
Вы можете запускать браузер в режиме киоска, для предотвращения выхода из него (например для Google Chrome при запуске можно указать " --kiosk --start-fullscreen")
P.S.:
- Друзья, если вас не затруднит, не убирайте, пожалуйста, мой копирайт внизу страницы. Вам это ничего не стоит, а мне приятно.
- Всё это реализовано без какого либо ТЗ и никто не верил в успех, поэтому пока что есть некоторая хаотичность, но постараюсь всё переделать более правильно =)
- Если вам помогло это решение, вы можете проспонсировать меня отправив слово "Спасибо". Ссылки на контакты ниже =)
- Если нужна помощь с внедрением, можем обсудить =).
{
"general": {
"debug": true,
// включить режим отладки
"log_path": "errors.log",
// путь к файлу журнала
"default_language": "ru",
// язык по умолчанию
"allow_unsafe_werkzeug": false,
// разрешить небезопасные операции в werkzeug
"button_change_theme": true,
// показать кнопку изменения темы
"button_fullscreen": true,
// показать кнопку перехода в полноэкранный режим
"button_backward": false,
// показать кнопку назад
"button_save_capture": false,
// показать кнопку сохранения кадра
"collapsed_keyboard": true
// показать клавиатуры свернутыми
},
"server": {
"host": "0.0.0.0",
// адрес сервера
"port": 80,
// порт сервера
"use_reloader": false,
// включить режим перезагрузки
"log_output": true,
// включить вывод журнала
"socketio_key": ""
// socketio ключ
},
"users": {
// логин:пароль по умолчанию admin:admin
"admin": "scrypt:32768:8:1$rsdPYhqaQqpXQQ0o$aa3359c86228b4cee5fe8c4ed694db4b371fa7fab5100fa7b446db7e1ed8077e3bb63228d4a1899aeeef9b8d15f8e8bdbcc3457f020bcb3ec320332c76b5896b"
// логин:пароль
},
"db": {
"uri": "sqlite:///system/database.db",
// подключение к базе данных
"prefix": ""
// префикс таблиц
},
"form": {
// конфигурация форм
"defect_show": true,
// показать форму брака
"correct_show": true,
// показать форму коррекции
"custom_fields": {
// конфигурация пользовательских полей
"field_one": {
"name": "field_one",
// название поля
"label": "Field One",
// подпись поля
"type": "text"
// тип поля
}
}
},
"detection_default": {
"video_show_scale": 50,
// масштаб вывода видео на странице
"video_show_quality": 50,
// качество вывода видео на странице
"video_fps": 0,
// ручная установка FPS (0 - автоматическая установка)
"weights_path": "yolo_cfg/models/yolov8n.pt",
// путь к модели Yolov8
"counting_area": [
[
0,
0
],
[
100,
0
],
[
100,
100
],
[
0,
100
]
],
// площадь подсчета (многоугольник)
"confidence": 0.7,
// порог доверия
"iou": 0.7,
// порог iou
"device": 0,
// указывает вычислительное устройство(а) для обучения (смотрите документацию ultralytics)
"vid_stride": 1,
// шаг видеопотока
"indicator_size": 10,
// размер индикатора
"counting_area_color": [
67,
211,
255
],
// цвет зоны подсчета
"classes": {}
// классы (объекты) для обнаружения (оставьте пустым для всех классов)
},
"detections": {
// конфигурации обнаружения
"ExampleCam": {
// Наименование подсчета (используется в адресе страницы, должно быть на латинице)
"label": "Label ExampleCam",
// Наименование подсчета (используется для вывода на страницах)
"start_total_count": 0,
// Число с которого начинается подсчет (по умолчанию 0, но если необходимо начать с какого-то числа, то можно указать)
"video_path": "",
// путь к видеофайлу или источнику камеры
"video_show_scale": 70,
// масштаб вывода видео на странице
"video_show_quality": 30,
// качество вывода видео на странице
"video_fps": 30,
// ручная установка FPS (необязательно)
"weights_path": "yolo_cfg/models/yolov8n.pt",
// путь к модели Yolov8
"counting_area": [
[
0,
0
],
[
100,
0
],
[
100,
100
],
[
0,
100
]
],
// площадь подсчета (многоугольник)
"confidence": 0.7,
// порог доверия
"iou": 0.7,
// порог iou
"device": 0,
// указывает вычислительное устройство(а) для обучения (смотрите документацию ultralytics)
"vid_stride": 1,
// шаг видеопотока
"indicator_size": 10,
// размер индикатора
"counting_area_color": [
255,
64,
0
],
// цвет зоны подсчета
"classes": {},
// классы (объекты) для обнаружения (оставьте пустым для всех классов)
"dataset_create": {
// автоматическое создание набора данных
"enable": true,
// включить создание набора данных
"probability": 0.05,
// вероятность создания изображения набора данных (число от 0.01 до 1, где 0.01 - 1% и 1 - 100%)
"path": "yolo_cfg/saved_images/ExampleCam"
// путь для сохранения набора данных
}
},
}
}
Александр Киреев
Website: https://bespredel.name
E-mail: hello@bespredel.name
GitHub: https://github.com/BespredeL
Ultralytics: https://github.com/ultralytics
AGPL-3.0 License: Эта OSI-approved лицензия с открытым исходным кодом идеально подходит для студентов и энтузиастов, способствуя открытому сотрудничеству и обмену знаниями.