weiyithu / SurroundOcc

[ICCV 2023] SurroundOcc: Multi-camera 3D Occupancy Prediction for Autonomous Driving
Apache License 2.0
771 stars 95 forks source link

Some problems I met when I started to train with mini-dataset #102

Open Apple-father opened 6 months ago

Apple-father commented 6 months ago

根据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进行手动的更改

Rogersljs commented 1 month ago

请问,你解决了吗?

Apple-father commented 1 month ago

我已收到