FCHD-Fully-Convolutional-Head-Detector

FCHD-Fully-Convolutional-Head-Detector copied to clipboard

run demo error

8

oints/head_detector_final test_image/test3.jpg checkpoints/head_detector_final (600.0, 789.7172236503856) WARNING:root:Setting up a new session... Traceback (most recent call last): File "head_detection_demo.py", line 73, in detect(args.img_path, args.model_path) File "head_detection_demo.py", line 51, in detect pred_bboxes_, _...

Oliverwang

运行demo 的时候： python head_detection_demo.py --img_path mov_021_178867.jpeg --model_path '/FCHD-Fully-Convolutional-Head-Detector/checkpoints/head_detector09191214_0.6141253457131631' 报错：RuntimeError: CUDA error: out of memory

tingnini

Is there a way to calculate the MAP in the code?

1

zhanghongsir

Does the writer set up CPU version? And how can I train and test on CPU?

kailaisun

why can't I change the invalid batch_size

2

when I change the invalid batch_size=128 or other number !=1, RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 861 and 811 in dimension 3...

KelvinHuang666

Bounding Boxes displacement error

16

I have run "head_detection_demo.py" and I haven't received good results. The bounding boxes seem correct but there is a kind of displacement respect to ground truth. 

pecia6

Can this demo be used to test video?

2

I want to count the number of heads in video,does this code can do it?Or do you know of any other relevant articles~~thanks

silencezx

Hi @aditya-vora How can I run demo on cpu ?. Is this only run on gpu ?? Thanks

Akhtar303

can I use resnet100 as pre-trainer model??anybody had tried for it?tell me please

KelvinHuang666

Hi When I ran demo so it's generate error RuntimeError: Error(s) in loading state_dict for Head_Detector_VGG16: Missing key(s) in state_dict: "extractor.0.weight", "extractor.0.bias", "extractor.2.weight", "extractor.2.bias", "extractor.5.weight", "extractor.5.bias", "extractor.7.weight", "extractor.7.bias", "extractor.10.weight", "extractor.10.bias",...

Akhtar303