Python3.7+
Python-OpenCV
Pytorch1.0+
PyQT5
python3 Main.py
“开始识别”
You will be able to see the person detected and the distance between the person and the camera from the picture.
At present, the model we are using is trained by Mobilenet2-SSD. If you want to change it into another model, you can modify it manually in the file Detect.py.
It should be noted that through our VOC test set, the evaluation of the mobilenet 2-ssd.pth model is about 67%, while that of the mobilenet 3-large-ssd.pth model is about 55%. We will make further improvements in the future.
We updated the interface, and now you can manually select the network to load, but if you want to modify and put it into your model, you need to implement it manually.
Make sure you choose the network before you start identifying it.