From ed7c7f48013731622bb51f4b0ac100cd204f7f25 Mon Sep 17 00:00:00 2001 From: Davi Date: Fri, 19 Jan 2024 15:42:21 -0300 Subject: [PATCH] Fixing ValueError 'inhomogeneous shape' due list-array conversion --- lart/datamodules/components/phalp_action_datatset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lart/datamodules/components/phalp_action_datatset.py b/lart/datamodules/components/phalp_action_datatset.py index 82afe2f..86d12dc 100644 --- a/lart/datamodules/components/phalp_action_datatset.py +++ b/lart/datamodules/components/phalp_action_datatset.py @@ -78,7 +78,7 @@ def __init__(self, opt, train=True): total_num_tracks=None) self.data = np.array(self.data) - self.track2video = np.array(self.track2video) + self.track2video = np.array(self.track2video, dtype=object) self.data = task_divider(self.data, self.opt.test_batch_id, self.opt.number_of_processes) self.track2video = task_divider(self.track2video, self.opt.test_batch_id, self.opt.number_of_processes)