crosstyan/OpenGait
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixing point cloud index none issue in collate_fn.py

Browse Source
This commit is contained in:
chuanfushen
2025-06-23 10:02:33 +08:00
committed by GitHub
parent 16a7c3f0bf
commit 347d157f9c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
opengait/data/collate_fn.py
+2 -2
View File
@@ -90,7 +90,7 @@ class CollateFn(object):
for i in range(feature_num): for i in range(feature_num):
for j in indices[:self.frames_all_limit] if self.frames_all_limit > -1 and len(indices) > self.frames_all_limit else indices: for j in indices[:self.frames_all_limit] if self.frames_all_limit > -1 and len(indices) > self.frames_all_limit else indices:
point_cloud_index = self.points_in_use.get('pointcloud_index') point_cloud_index = self.points_in_use.get('pointcloud_index') if self.points_in_use else None
if self.points_in_use is not None and point_cloud_index is not None and i == point_cloud_index: if self.points_in_use is not None and point_cloud_index is not None and i == point_cloud_index:
points_num = self.points_in_use.get('points_num') points_num = self.points_in_use.get('points_num')
sample_points = (random.choices(range(len(seqs[i][j])), k=points_num) sample_points = (random.choices(range(len(seqs[i][j])), k=points_num)
@@ -121,4 +121,4 @@ class CollateFn(object):
batch[-1] = np.asarray(seqL_batch) batch[-1] = np.asarray(seqL_batch)
batch[0] = fras_batch batch[0] = fras_batch
return batch return batch