Spaces:
Running
on
Zero
Running
on
Zero
Update stf_utils.py
Browse files- 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'])
|