Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

hi, python head_detection_demo.py have error #58

Open
zhanghongruiupup opened this issue Jan 19, 2020 · 1 comment
Open

hi, python head_detection_demo.py have error #58

zhanghongruiupup opened this issue Jan 19, 2020 · 1 comment

Comments

@zhanghongruiupup
Copy link

python head_detection_demo.py --img_path /home/fwq/pic/1.png --model_path /home/fwq/zhanghongruils/head/fchd/model/head_detector_final

info:
the python code for non_maximum_suppression is about 2x slow
It is strongly recommended to build cython code:
cd model/utils/nms/; python3 build.py build_ext --inplace cd model/utils/nms/; python3 build.py build_ext --inplace''')
Setting up a new session...
Traceback (most recent call last):
File "head_detection_demo.py", line 69, in
detect(args.img_path, args.model_path)
File "head_detection_demo.py", line 49, in detect
pred_bboxes_, _ = head_detector.predict(img, scale, mode='evaluate', thresh=THRESH)
File "/home/fwq/zhanghongruils/head/fchd/src/head_detector.py", line 72, in predict
_, _, rois, rois_scores, _ = self.forward(x, scale=scale)
File "/home/fwq/zhanghongruils/head/fchd/src/head_detector.py", line 26, in forward
rpn_locs, rpn_scores, rois, rois_scores, anchor = self.rpn(h, img_size, scale)
File "/home/fwq/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 547, in call
result = self.forward(*input, **kwargs)
File "/home/fwq/zhanghongruils/head/fchd/src/region_proposal_network.py", line 79, in forward
scale=scale
File "/home/fwq/zhanghongruils/head/fchd/src/creator_tool.py", line 150, in call
roi = roi[keep,:]
ValueError: object array method not producing an array

hope your reply

@kimsent
Copy link

kimsent commented Feb 1, 2021

can you please upload the weights? i cannot download it.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants