Open ranjiewwen opened 5 years ago
i run the infer_simple.py too. but i met another error, have u ever gotton this error:
Called with args: Namespace(cfg_file='configs/panet/e2e_panet_R-50-FPN_2x_mask.yaml', cuda=True, dataset='coco2017', image_dir='demo/sample_images', images=None, load_ckpt='panet_mask_step179999.pth', load_detectron=None, merge_pdfs=True, output_dir='infer_outputs', set_cfgs=[]) load cfg from file: configs/panet/e2e_panet_R-50-FPN_2x_mask.yaml /home/cm/PANet/lib/core/config.py:1037: YAMLLoadWarning: calling yaml.load() without Loader=... is deprecated, as the default Loader is unsafe. Please read https://msg.pyyaml.org/load for full details. yaml_cfg = AttrDict(yaml.load(f)) loading checkpoint panet_mask_step179999.pth img 0 THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=844 error=11 : invalid argument car 0.97782826 Traceback (most recent call last): File "tools/infer_simple.py", line 176, in <module> main() File "tools/infer_simple.py", line 163, in main kp_thresh=2 File "/home/cm/PANet/lib/utils/vis.py", line 183, in vis_one_image e.copy(), cv2.RETR_CCOMP, cv2.CHAIN_APPROX_NONE) ValueError: not enough values to unpack (expected 3, got 2)