kexinyi / ns-vqa

Neural-symbolic visual question answering
255 stars 63 forks source link

(Object Detection) The size of tensor a (75) must match the size of tensor b (76) at non-singleton dimension 3 #6

Open theresequieta opened 5 years ago

theresequieta commented 5 years ago

I'm trying to run pretrained models. I'm in Step 1: object detection. I encounter this error: Traceback (most recent call last): File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/tools/test_net.py", line 128, in check_expected_results=True) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/core/test_engine.py", line 150, in run_inference all_results = result_getter() File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/core/test_engine.py", line 130, in result_getter multi_gpu=multi_gpu_testing File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/core/test_engine.py", line 180, in test_net_on_dataset args, dataset_name, proposal_file, output_dir, gpu_id=gpu_id File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/core/test_engine.py", line 275, in test_net cls_boxes_i, cls_segms_i, cls_keyps_i = im_detect_all(model, im, box_proposals, timers) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/core/test.py", line 71, in im_detect_all model, im, cfg.TEST.SCALE, cfg.TEST.MAX_SIZE, box_proposals) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/core/test.py", line 152, in im_detect_bbox return_dict = model(inputs) File "/home/therese/anaconda2/envs/ns-vqa/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(*input, *kwargs) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/nn/parallel/data_parallel.py", line 108, in forward outputs = [self.module(inputs[0], kwargs[0])] File "/home/therese/anaconda2/envs/ns-vqa/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, kwargs) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/modeling/model_builder.py", line 157, in forward return self._forward(data, im_info, roidb, rpn_kwargs) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/modeling/model_builder.py", line 168, in _forward blob_conv = self.Conv_Body(im_data) File "/home/therese/anaconda2/envs/ns-vqa/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, *kwargs) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/modeling/FPN.py", line 246, in forward self.topdown_lateral_modules[i](fpn_inner_blobs[-1], conv_body_blobs[-(i+2)]) File "/home/therese/anaconda2/envs/ns-vqa/lib/python3.6/site-packages/torch/nn/modules/module.py", line 491, in call result = self.forward(input, **kwargs) File "/home/therese/PycharmProjects/neural_symbolic_vqa/ns-vqa/scene_parse/mask_rcnn/lib/modeling/FPN.py", line 310, in forward return lat + td RuntimeError: The size of tensor a (75) must match the size of tensor b (76) at non-singleton dimension 3

Anyone has encountered this before? My pytorch version is 0.4.0