YOLO로 사람 수 세기
YOLO People Counting
GitHub : https://github.com/jjxxmiin/People_counting_yolo
구성
- raspberry pi 3 (rasbian)
- webcam
- NCS2
requirement
- openvino toolkit
- opencvs
- sklearn
- filterpy
- imutils
- sort
필독!!
SORT requirement 설치 (numba 설치가 안될 때 보세요)
1
2
3
sudo apt install libblas-dev llvm python3-pip python3-scipy
pip3 install llvmlite==0.15.0
pip3 install numba==0.30.1
중간에 numba 0.30.1을 설치할 때 계속 llvmlite 0.28.0을 설치하려는 경향 때문에 오류가 계속 발생했다. 만약 이런오류가 발생한다면 여기 에 들어가서 받고
1
2
3
cd numba-0.30.1
python setup.py build_ext --inplace
python setup.py install
- 중간에 pip을 덮어서.. pip3가 망가짐 그래서 아래와 같이 사용
1
python3 -m pip ~
- 남은 requirement 설치
1
python3 -m pip install sklearn tqdm imutils
사용법
- 위에 xml_path와 bin_path를 xml,bin 파일 경로로 바꾸어준다.
- capture function에서 아래 코드 주석처리
1
# cv2.imwrite('test2.jpg',frame)
- 스케쥴러를 이용해서 웹캠으로 10초에 한번씩 사람의 수를 counting한다.
1
python3 main.py
logger
- 인터넷이 연결되어있는지 확인 -> 서버로 전송가능
- 카메라가 연결되어있는지 확인
Citation
YOLO
1
2
3
4
5
6
@article{redmon2016yolo9000,
title={YOLO9000: Better, Faster, Stronger},
author={Redmon, Joseph and Farhadi, Ali},
journal={arXiv preprint arXiv:1612.08242},
year={2016}
}
SORT
1
2
3
4
5
6
7
8
9
@inproceedings{Bewley2016_sort,
author={Bewley, Alex and Ge, Zongyuan and Ott, Lionel and Ramos, Fabio and Upcroft, Ben},
booktitle={2016 IEEE International Conference on Image Processing (ICIP)},
title={Simple online and realtime tracking},
year={2016},
pages={3464-3468},
keywords={Benchmark testing;Complexity theory;Detectors;Kalman filters;Target tracking;Visualization;Computer Vision;Data Association;Detection;Multiple Object Tracking},
doi={10.1109/ICIP.2016.7533003}
}
Reference
This post is licensed under CC BY 4.0 by the author.