fix

wshanon · Aug 6, 2019 · 7a7c394 · 7a7c394
1 parent f225227
commit 7a7c394
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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)