search

roytseng-tw / Detectron.pytorch

A pytorch implementation of Detectron. Both training from scratch and inferring directly from pretrained Detectron weights are available.
MIT License
2.82k stars 565 forks source link

test error of keypoints #159

Closed youngwanLEE closed 6 years ago

youngwanLEE commented 6 years ago 
python tools/test_net.py --cfg configs/baselines/e2e_keypoint_rcnn_R-50-FPN_1x.yaml --load_ckpt Outputs/e2e_keypoint_rcnn_R-50-FPN_1x/Oct08-12-20-44_xp5_step/ckpt/model_step2882.pth --dataset coco2017 --multi-gpu-testing

When I test e2e_keypoint_rcnn_R-50-FPN as above, this error message occurs. image

I guess this error occurs due to fast_rcnn_heads.roi_2mlp_head in here.

In addition,

When I try to train e2e_keypoint_rcnn_R-101-FPN_1x by using the yaml, I can't find head_builder.roi_2mlp_head function and head_builder file.

System information

kaminocode commented 5 years ago

Even I'm facing the same issue. Whats the error?