The purpose of this project is to build a system to detect people pose in a video.
- First you need to download docker and run docker, install miniconda and install cuda
- Clone this repo
- Now, install Kafka:
pip install confluent-kafka
- After that, create conda environment, install MMEngine, MMCV, MMPOSE using MIM, BoxMOT and install ultralytics:
conda create --name DNP python=3.8 -y
pip install -U openmim
mim install mmengine
mim install "mmcv>=2.0.1"
mim install "mmdet>=3.1.0"
mim install "mmpose>=1.1.0"
conda install -c conda-forge ultralytics
pip install boxmot
- Download the config file:
mim download mmpose --config td-hm_hrnet-w48_8xb32-210e_coco-256x192 --dest .
- Run each block in different commandline windows:
conda activate DNP
docker-compose up
conda activate DNP
python createKafkaTopic.py
python predictPose.py
conda activate DNP
python predictBoundingBox.py
conda activate DNP
python sendVideo.py