Update app.py
Browse files
app.py
CHANGED
@@ -201,6 +201,10 @@ def infer():
|
|
201 |
img = img.resize((960, 520))
|
202 |
# display the PIL image
|
203 |
#img.show()
|
|
|
|
|
|
|
|
|
204 |
img.save('frame_input.jpg')
|
205 |
#res = get_warp_res('frame_input.jpg', "predicted_flow.jpg", 'warped.png')
|
206 |
#print(res)
|
|
|
201 |
img = img.resize((960, 520))
|
202 |
# display the PIL image
|
203 |
#img.show()
|
204 |
+
frame1pil = np.array(img.convert('RGB'))
|
205 |
+
print(f"frame1pil: {frame1pil}")
|
206 |
+
print(f"frame1pil shape: {frame1pil.shape}")
|
207 |
+
print(f"frame1pil dtype: {frame1pil.dtype}")
|
208 |
img.save('frame_input.jpg')
|
209 |
#res = get_warp_res('frame_input.jpg', "predicted_flow.jpg", 'warped.png')
|
210 |
#print(res)
|