根据data.md文档我先使用nuscenes-mini数据集运行:
python generate_occupancy_nuscenes.py --config_path ./config.yaml --label_mapping ./nuscenes.yaml --split [train/val] --save_path [your/save/path]
生成的npy文件按格式都保存到了./data/nuscenes_occ/samples/下,
然后我进行训练:
./tools/dist_train.sh ./projects/configs/surroundocc/surroundocc.py 1 ./work_dirs/surroundocc
出现了如下的报错:
Traceback (most recent call last):
File "./tools/train.py", line 261, in
main()
File "./tools/train.py", line 257, in main
meta=meta)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/surroundocc/apis/train.py", line 35, in custom_train_model
meta=meta)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/surroundocc/apis/mmdet_train.py", line 201, in custom_train_detector
runner.run(data_loaders, cfg.workflow)
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 127, in run
epoch_runner(data_loaders[i], **kwargs)
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 47, in train
for i, data_batch in enumerate(self.data_loader):
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/datasets/nuscenes_occupancy_dataset.py", line 132, in getitem
data = self.prepare_train_data(idx)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/datasets/nuscenes_occupancy_dataset.py", line 47, in prepare_train_data
example = self.pipeline(input_dict)
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmdet/datasets/pipelines/compose.py", line 40, in call
data = t(data)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/datasets/pipelines/loading.py", line 28, in call
occ = np.load(results['occ_path'])
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/numpy/lib/npyio.py", line 416, in load
fid = stack.enter_context(open(os_fspath(file), "rb"))
FileNotFoundError: [Errno 2] No such file or directory: './data/nuscenes_occ/samples/n015-2018-11-21-19-38-26+0800__LIDAR_TOP__1542800382897959.pcd.bin'
我不知道是否需要对occ_path进行手动的更改
根据data.md文档我先使用nuscenes-mini数据集运行:
main()
File "./tools/train.py", line 257, in main
meta=meta)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/surroundocc/apis/train.py", line 35, in custom_train_model
meta=meta)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/surroundocc/apis/mmdet_train.py", line 201, in custom_train_detector
runner.run(data_loaders, cfg.workflow)
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 127, in run
epoch_runner(data_loaders[i], **kwargs)
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmcv/runner/epoch_based_runner.py", line 47, in train
for i, data_batch in enumerate(self.data_loader):
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 521, in next
data = self._next_data()
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/dataloader.py", line 561, in _next_data
data = self._dataset_fetcher.fetch(index) # may raise StopIteration
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/torch/utils/data/_utils/fetch.py", line 49, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/datasets/nuscenes_occupancy_dataset.py", line 132, in getitem
data = self.prepare_train_data(idx)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/datasets/nuscenes_occupancy_dataset.py", line 47, in prepare_train_data
example = self.pipeline(input_dict)
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/mmdet/datasets/pipelines/compose.py", line 40, in call
data = t(data)
File "/root/autodl-tmp/SurroundOcc/projects/mmdet3d_plugin/datasets/pipelines/loading.py", line 28, in call
occ = np.load(results['occ_path'])
File "/root/autodl-tmp/miniconda3/envs/surroundocc/lib/python3.7/site-packages/numpy/lib/npyio.py", line 416, in load
fid = stack.enter_context(open(os_fspath(file), "rb"))
FileNotFoundError: [Errno 2] No such file or directory: './data/nuscenes_occ/samples/n015-2018-11-21-19-38-26+0800__LIDAR_TOP__1542800382897959.pcd.bin'
我不知道是否需要对occ_path进行手动的更改
python generate_occupancy_nuscenes.py --config_path ./config.yaml --label_mapping ./nuscenes.yaml --split [train/val] --save_path [your/save/path]
生成的npy文件按格式都保存到了./data/nuscenes_occ/samples/下, 然后我进行训练: ./tools/dist_train.sh ./projects/configs/surroundocc/surroundocc.py 1 ./work_dirs/surroundocc 出现了如下的报错: Traceback (most recent call last): File "./tools/train.py", line 261, in