YonghaoHe / LFFD-A-Light-and-Fast-Face-Detector-for-Edge-Devices

A light and fast one class detection framework for edge devices. We provide face detector, head detector, pedestrian detector, vehicle detector......
MIT License
1.32k stars 329 forks source link

face detection demo.py run error #38

Closed ForestWang closed 5 years ago

ForestWang commented 5 years ago

Hi : when i run face detection demo.py, got the following error:

Traceback (most recent call last): File "demo.py", line 138, in main() File "demo.py", line 68, in main NMS_threshold=0.4, NMS_flag=True, skip_scale_branch_list=[]) File "../accuracy_evaluation/predict.py", line 173, in predict score_map = numpy.squeeze(outputs[i * 2], (0, 1)) File "/usr/local/lib/python3.5/dist-packages/numpy/core/fromnumeric.py", line 1388, in squeeze return squeeze(axis=axis)