Setting class attribute final_result

joris-gentinetta · Sep 7, 2022 · 0574e08 · 0574e08
1 parent 308cc9a
commit 0574e08
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/alphapose/utils/writer.py b/alphapose/utils/writer.py
@@ -78,7 +78,7 @@ def start(self):
         return self
 
     def update(self):
-        final_result = []
+        self.final_result = []
         norm_type = self.cfg.LOSS.get('NORM_TYPE', None)
         hm_size = self.cfg.DATA_PRESET.HEATMAP_SIZE
         if self.save_video:
@@ -100,7 +100,7 @@ def update(self):
                 # if the thread indicator variable is set (img is None), stop the thread
                 if self.save_video:
                     stream.release()
-                write_json(final_result, self.opt.outputpath, form=self.opt.format, for_eval=self.opt.eval)
+                write_json(self.final_result, self.opt.outputpath, form=self.opt.format, for_eval=self.opt.eval)
                 print("Results have been written to json.")
                 return
             # image channel RGB->BGR
@@ -168,7 +168,7 @@ def update(self):
                     for i in range(len(poseflow_result)):
                         result['result'][i]['idx'] = poseflow_result[i]['idx']
 
-                final_result.append(result)
+                self.final_result.append(result)
                 if self.opt.save_img or self.save_video or self.opt.vis:
                     if hm_data.size()[1] == 49:
                         from alphapose.utils.vis import vis_frame_dense as vis_frame
@@ -224,7 +224,7 @@ def clear(self, queue):
 
     def results(self):
         # return final result
-        print(self.final_result)
+        # print(self.final_result)
         return self.final_result
 
     def recognize_video_ext(self, ext=''):