Repository navigation
Problems when running the demo #15
Copy link
Copy link
Open
Description
@zyc10ud Thank you for your excellent work!
I want to run demo by the following command
/home/wll/anaconda3/envs/vma/bin/python demo/custom_infer.py \
/home/wll/Project/VMA/demo/image_data \
projects/configs/vma_icurb.py \
ckpts/sd_line.pth \
--trajectory_sample_num 15 \
--element_type line \
--out_dir output \
--visualize False
The following error message appears. I did not crop or modify the input image. Is there something wrong with my execution steps or commands?
Traceback (most recent call last):
File "demo/custom_infer.py", line 64, in <module>
main()
File "demo/custom_infer.py", line 60, in main
infer_combine_function(args, data_list)
File "/home/wll/Project/VMA/demo/inference_and_combine_function.py", line 223, in driving_line_infer_and_save
results_list, left_top = inference_detector_forcurb_bigimg(model, get_sub_data_driving, upload_img_path, trajectory_path, args.trajectory_sample_num, get_traj_pad=True, pad=False, attr=True)
File "/home/wll/Project/VMA/demo/inference_and_combine_function.py", line 175, in inference_detector_forcurb_bigimg
result = model(return_loss=False, rescale=True, **data)
File "/home/wll/anaconda3/envs/vma/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "/home/wll/Project/VMA/projects/mmdet3d_plugin/vma/detectors/vma.py", line 114, in forward
return self.forward_test(**kwargs)
File "/home/wll/Project/VMA/projects/mmdet3d_plugin/vma/detectors/vma.py", line 158, in forward_test
bbox_results = self.simple_test(
File "/home/wll/Project/VMA/projects/mmdet3d_plugin/vma/detectors/vma.py", line 219, in simple_test
img_feats = self.extract_feat(img=img, img_metas=img_metas)
File "/home/wll/anaconda3/envs/vma/lib/python3.8/site-packages/mmcv/runner/fp16_utils.py", line 98, in new_func
return old_func(*args, **kwargs)
File "/home/wll/Project/VMA/projects/mmdet3d_plugin/vma/detectors/vma.py", line 69, in extract_feat
img_feats = self.extract_img_feat(img)
File "/home/wll/Project/VMA/projects/mmdet3d_plugin/vma/detectors/vma.py", line 53, in extract_img_feat
img_feats = self.img_backbone(img)
File "/home/wll/anaconda3/envs/vma/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "/home/wll/Project/VMA/projects/mmdet3d_plugin/vma/backbones/icurb_backbone.py", line 66, in forward
x = F.relu(self.bn1(self.conv1(x)))
File "/home/wll/anaconda3/envs/vma/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1051, in _call_impl
return forward_call(*input, **kwargs)
File "/home/wll/anaconda3/envs/vma/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 443, in forward
return self._conv_forward(input, self.weight, self.bias)
File "/home/wll/anaconda3/envs/vma/lib/python3.8/site-packages/torch/nn/modules/conv.py", line 439, in _conv_forward
return F.conv2d(input, weight, bias, self.stride,
RuntimeError: Given groups=1, weight of size [64, 4, 7, 7], expected input[1, 3, 1000, 1000] to have 4 channels, but got 3 channels instead
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels
Activity
iwander-all commentedon Jul 24, 2024
the same, none of the commands in demo readme is played successfully in my try
iwander-all commentedon Jul 25, 2024
@Lin239522 终于玩出来了,指令是这样的:
`python ./demo/custom_infer.py ./demo/image_data/ ./projects/configs/vma_res152_e80_line.py ./ckpts/sd_line.pth --trajectory_sample_num 2 --element_type line --out_dir ./demo/out/ --visualize True
python ./demo/custom_infer.py ./demo/image_data/ ./projects/configs/vma_res152_e80_box.py ./ckpts/sd_box.pth --trajectory_sample_num 2 --element_type box --out_dir ./demo/out/ --visualize True
python ./demo/custom_infer.py ./demo/image_data/ ./projects/configs/vma_icurb_backbone_e80_freespace.py ./ckpts/sd_freespace.pth --trajectory_sample_num 2 --element_type freespace --out_dir ./demo/out/ --visualize True`
咋说呢,预测的不是很准啊....
Lin239522 commentedon Jul 26, 2024
输入应该是图像+json? 请问你知道 demo里面的 (demo/trajectory_data/sample_1.json) 是怎么得到的吗
iwander-all commentedon Jul 26, 2024
我也不造...只是跑通了这个demo,太磨人了
lonelyriki commentedon Aug 15, 2024
这个是轨迹数据啊,是做数据的人自己处理的.....