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'])
|