deepvo 的preprocess文件,文档说会有6个浮点数表示的数据,但是我运行后一行数据有15个,而且在执行main文件后报错:
Traceback (most recent call last):
File "F:\yjw\deepvo\DeepVO-pytorch-master\main.py", line 116, in <module>
for _, t_x, t_y in train_dl:
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\utils\data\dataloader.py", line 530, in __next__
data = self._next_data()
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\utils\data\dataloader.py", line 1224, in _next_data
return self._process_data(data)
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\utils\data\dataloader.py", line 1250, in _process_data
data.reraise()
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\_utils.py", line 457, in reraise
raise exception
ValueError: Caught ValueError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\utils\data\_utils\worker.py", line 287, in _worker_loop
data = fetcher.fetch(index)
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "C:\ProgramData\Anaconda3\envs\ros_env\lib\site-packages\torch\utils\data\_utils\fetch.py", line 49, in <listcomp>
data = [self.dataset[idx] for idx in possibly_batched_index]
File "F:\yjw\deepvo\DeepVO-pytorch-master\data_helper.py", line 201, in __getitem__
groundtruth_rotation = raw_groundtruth[1][0].reshape((3, 3)).T # opposite rotation of the first frame
ValueError: cannot reshape array of size 0 into shape (3,3)
Process finished with exit code 1
请问怎么改?