diff --git a/processor/demo_offline.py b/processor/demo_offline.py index 3a299e052..45bf12c92 100644 --- a/processor/demo_offline.py +++ b/processor/demo_offline.py @@ -75,7 +75,6 @@ def predict(self, data): for t in range(num_frame): frame_label_name = list() for m in range(num_person): - print(output.shape) person_label = output[:, t, :, m].sum(dim=1).argmax(dim=0) person_label_name = self.label_name[person_label] frame_label_name.append(person_label_name)