This project is about videosources emergency detection made with ZeroMQ infrastructure
- ZeroMQ
- YOLOv8
- PSNR
- OpenCV
- Multiprocessing & Threading
- Torch with CUDA
- Python3.12
pip install -r requirements.txt- Download all necessary files from Google Drive :
- (optional) place test videos & meanframes to
./videosfolder - place YOLO models to
./modelsfolder - set up
config.jsonwith actual files locations, ip addresses & ports, psnr threshold
Notes:
meanframes must have same resolution as original videos, '.jpg' hardcoded
meanframes were created on "normal" parts of videos withpsnr.py
Create meanframes for PSNR function:
- Prepare "normal" videofragment (not containing emergency situations) in video editor
- Use
psnr.py:python psnr.py <videopath>
- Meanframe will be created in same directory with same filename in '.jpg' format
After you've set up config.json, just:
python run.pyCtrl+C to stop
- create meanframes automatically
- train models on another datasets
- add resolution to config
- Ctrl+C handler
- Auto scale processes
- Test on realtime videosources
