A flexible and extensible framework for gait recognition. You can focus on designing your own models and comparing with state-of-the-arts easily with the help of OpenGait.
664
stars
154
forks
source link
A Generat Heatmap error occurs during data set processing #206
Dear Professor,
I am replicating Skeletongai ++ using the GREW data set，has the following problems：
A Generat Heatmap error occurs during data set processing. The error message is as follows
Traceback (most recent call last):
File "datasets/pretreatmentheatmap.py", line 709, in
for , tmp in tqdm(enumerate(dataloader), total=len(dataloader)):
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/tqdm/std.py", line 1182, in iter
for obj in iterable:
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 628, in next
data = self._next_data()
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1333, in _next_data
return self._process_data(data)
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1359, in _process_data
data.reraise()
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/_utils.py", line 543, in reraise
raise exception
ValueError: Caught ValueError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop
data = fetcher.fetch(index)
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 58, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 58, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "datasets/pretreatment_heatmap.py", line 634, in getitem
pose_data = pose_data[:,2:].reshape(-1, 17, 3)
ValueError: cannot reshape array of size 2925 into shape (17,3)
Could you help me see what's wrong.Thank you very much and look forward to your early reply.
github-actions[bot]
commented
1 week ago
Dear Professor, I am replicating Skeletongai ++ using the GREW data set，has the following problems： A Generat Heatmap error occurs during data set processing. The error message is as follows Traceback (most recent call last): File "datasets/pretreatmentheatmap.py", line 709, in
for , tmp in tqdm(enumerate(dataloader), total=len(dataloader)):
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/tqdm/std.py", line 1182, in iter
for obj in iterable:
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 628, in next
data = self._next_data()
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1333, in _next_data
return self._process_data(data)
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/dataloader.py", line 1359, in _process_data
data.reraise()
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/_utils.py", line 543, in reraise
raise exception
ValueError: Caught ValueError in DataLoader worker process 0.
Original Traceback (most recent call last):
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/_utils/worker.py", line 302, in _worker_loop
data = fetcher.fetch(index)
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 58, in fetch
data = [self.dataset[idx] for idx in possibly_batched_index]
File "/root/miniconda3/envs/AllinOne/lib/python3.8/site-packages/torch/utils/data/_utils/fetch.py", line 58, in
data = [self.dataset[idx] for idx in possibly_batched_index]
File "datasets/pretreatment_heatmap.py", line 634, in getitem
pose_data = pose_data[:,2:].reshape(-1, 17, 3)
ValueError: cannot reshape array of size 2925 into shape (17,3)
Could you help me see what's wrong.Thank you very much and look forward to your early reply.