如何在nuscenes的mini数据集上跑出可视化结果

[zhaojiankuan]

我想用mini数据集测试下maptr的效果，想熟悉下跑通的流程，但是遇到了一些问题
运行命令是：python -m torch.distributed.launch --nproc_per_node=1 --master_port 12345 tools/test.py
在for i, data in enumerate(data_loader):报错 EOFError: Ran out of input TypeError: cannot pickle 'dict_keys' object

NOTE: Redirects are currently not supported in Windows or MacOs.
D:\anaconda\envs\bev\lib\site-packages\torch\distributed\launch.py:178: FutureWarning: The module torch.distributed.launch is deprecated
and will be removed in future. Use torch.distributed.run.
Note that --use_env is set by default in torch.distributed.run.
If your script expects --local_rank argument to be set, please
change it to read from os.environ['LOCAL_RANK'] instead. See
https://pytorch.org/docs/stable/distributed.html#launch-utility for
further instructions

warnings.warn(
projects.mmdet3d_plugin
E:\code/MapTR\projects\mmdet3d_plugin\bevformer\modules\custom_base_transformer_layer.py:94: UserWarning: The arguments feedforward_channels in BaseTransformerLayer has been deprecated, now you should se
t feedforward_channels and other FFN related arguments to a dict named ffn_cfgs.
warnings.warn(
E:\code/MapTR\projects\mmdet3d_plugin\bevformer\modules\custom_base_transformer_layer.py:94: UserWarning: The arguments ffn_dropout in BaseTransformerLayer has been deprecated, now you should set ffn_dr op and other FFN related arguments to a dict named ffn_cfgs.
warnings.warn(
E:\code/MapTR\projects\mmdet3d_plugin\bevformer\modules\custom_base_transformer_layer.py:94: UserWarning: The arguments ffn_num_fcs in BaseTransformerLayer has been deprecated, now you should set num_fc s and other FFN related arguments to a dict named ffn_cfgs.
warnings.warn(
D:\anaconda\envs\bev\lib\site-packages\mmcv\cnn\bricks\transformer.py:341: UserWarning: The arguments feedforward_channels in BaseTransformerLayer has been deprecated, now you should set feedforward_cha nnels and other FFN related arguments to a dict named ffn_cfgs.
warnings.warn(
D:\anaconda\envs\bev\lib\site-packages\mmcv\cnn\bricks\transformer.py:341: UserWarning: The arguments ffn_dropout in BaseTransformerLayer has been deprecated, now you should set ffn_drop and other FFN
related arguments to a dict named ffn_cfgs.
warnings.warn(
D:\anaconda\envs\bev\lib\site-packages\mmcv\cnn\bricks\transformer.py:341: UserWarning: The arguments ffn_num_fcs in BaseTransformerLayer has been deprecated, now you should set num_fcs and other FFN r
elated arguments to a dict named ffn_cfgs.
warnings.warn(
D:\anaconda\envs\bev\lib\site-packages\mmcv\cnn\bricks\transformer.py:92: UserWarning: The arguments dropout in MultiheadAttention has been deprecated, now you can separately set attn_drop(float), proj
_drop(float), and dropout_layer(dict)
warnings.warn('The arguments dropout in MultiheadAttention '
load checkpoint from local path: E:/code/MapTR/ckpts/maptr_nano_r18_110e.pth
The model and loaded state dict do not match exactly

unexpected key in source state_dict: pts_bbox_head.transformer.encoder.layers.0.attentions.1.attention.grid_offsets

[ ] 0/81, elapsed: 0s, ETA:Traceback (most recent call last):
File "tools/test.py", line 269, in
main()
File "tools/test.py", line 240, in main
Traceback (most recent call last):
File "", line 1, in
outputs = custom_multi_gpu_test(model, data_loader, args.tmpdir,
File "D:\anaconda\envs\bev\lib\multiprocessing\spawn.py", line 116, in spawn_main
File "E:\code/MapTR\projects\mmdet3d_plugin\bevformer\apis\test.py", line 70, in custom_multi_gpu_test
exitcode = _main(fd, parent_sentinel)
File "D:\anaconda\envs\bev\lib\multiprocessing\spawn.py", line 126, in _main
self = reduction.pickle.load(from_parent)
EOFError: Ran out of input
for i, data in enumerate(data_loader):
File "D:\anaconda\envs\bev\lib\site-packages\torch\utils\data\dataloader.py", line 359, in iter
return self._get_iterator()
File "D:\anaconda\envs\bev\lib\site-packages\torch\utils\data\dataloader.py", line 305, in _get_iterator
return _MultiProcessingDataLoaderIter(self)
File "D:\anaconda\envs\bev\lib\site-packages\torch\utils\data\dataloader.py", line 918, in init
w.start()
File "D:\anaconda\envs\bev\lib\multiprocessing\process.py", line 121, in start
self._popen = self._Popen(self)
File "D:\anaconda\envs\bev\lib\multiprocessing\context.py", line 224, in _Popen
return _default_context.get_context().Process._Popen(process_obj)
File "D:\anaconda\envs\bev\lib\multiprocessing\context.py", line 327, in _Popen
return Popen(process_obj)
File "D:\anaconda\envs\bev\lib\multiprocessing\popen_spawn_win32.py", line 93, in init
reduction.dump(process_obj, to_child)
File "D:\anaconda\envs\bev\lib\multiprocessing\reduction.py", line 60, in dump
ForkingPickler(file, protocol).dump(obj)
TypeError: cannot pickle 'dict_keys' object
ERROR:torch.distributed.elastic.multiprocessing.api:failed (exitcode: 1) local_rank: 0 (pid: 25492) of binary: D:\anaconda\envs\bev\python.exe
Traceback (most recent call last):
File "D:\anaconda\envs\bev\lib\runpy.py", line 194, in _run_module_as_main
return _run_code(code, main_globals, None,
File "D:\anaconda\envs\bev\lib\runpy.py", line 87, in _run_code
exec(code, run_globals)
File "D:\anaconda\envs\bev\lib\site-packages\torch\distributed\launch.py", line 193, in
main()
File "D:\anaconda\envs\bev\lib\site-packages\torch\distributed\launch.py", line 189, in main
launch(args)
File "D:\anaconda\envs\bev\lib\site-packages\torch\distributed\launch.py", line 174, in launch
run(args)
File "D:\anaconda\envs\bev\lib\site-packages\torch\distributed\run.py", line 689, in run
elastic_launch(
File "D:\anaconda\envs\bev\lib\site-packages\torch\distributed\launcher\api.py", line 116, in call
return launch_agent(self._config, self._entrypoint, list(args))
File "D:\anaconda\envs\bev\lib\site-packages\torch\distributed\launcher\api.py", line 244, in launch_agent
raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError:

---

      tools/test.py FAILED

=======================================
Root Cause:
[0]:
time: 2024-03-14_14:21:03
rank: 0 (local_rank: 0)
exitcode: 1 (pid: 25492)
error_file: <N/A>
msg: "Process failed with exitcode 1"

Other Failures:
<NO_OTHER_FAILURES>

---