yerang commited on
Commit
a8e22e9
·
verified ·
1 Parent(s): c1a3728

Update stf_utils.py

Browse files
Files changed (1) hide show
  1. stf_utils.py +1 -0
stf_utils.py CHANGED
@@ -159,6 +159,7 @@ class STFPipeline:
159
  )
160
  for idx, (it, chunk) in enumerate(gen_infer, pivot):
161
  frame = next(reader)
 
162
  composed = self.template.compose(idx, frame, it)
163
  frame_name = f"{idx}".zfill(5)+".jpg"
164
  results.append(it['pred'])
 
159
  )
160
  for idx, (it, chunk) in enumerate(gen_infer, pivot):
161
  frame = next(reader)
162
+ print('frame=', frame.shape, frame[:10])
163
  composed = self.template.compose(idx, frame, it)
164
  frame_name = f"{idx}".zfill(5)+".jpg"
165
  results.append(it['pred'])