Spaces:
Paused
Paused
fix film bugs
Browse files- libs/film/predict.py +4 -2
libs/film/predict.py
CHANGED
|
@@ -34,11 +34,13 @@ class Predictor(cog.BasePredictor):
|
|
| 34 |
# make sure 2 images are the same size
|
| 35 |
#img1 = Image.open(str(frame1))
|
| 36 |
#img2 = Image.open(str(frame2))
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
if not img1.size == img2.size:
|
| 38 |
img1 = img1.crop((0, 0, min(img1.size[0], img2.size[0]), min(img1.size[1], img2.size[1])))
|
| 39 |
img2 = img2.crop((0, 0, min(img1.size[0], img2.size[0]), min(img1.size[1], img2.size[1])))
|
| 40 |
-
frame1 = 'new_frame1.png'
|
| 41 |
-
frame2 = 'new_frame2.png'
|
| 42 |
img1.save(frame1)
|
| 43 |
img2.save(frame2)
|
| 44 |
|
|
|
|
| 34 |
# make sure 2 images are the same size
|
| 35 |
#img1 = Image.open(str(frame1))
|
| 36 |
#img2 = Image.open(str(frame2))
|
| 37 |
+
frame1 = 'new_frame1.png'
|
| 38 |
+
frame2 = 'new_frame2.png'
|
| 39 |
+
img1.save(frame1)
|
| 40 |
+
img2.save(frame2)
|
| 41 |
if not img1.size == img2.size:
|
| 42 |
img1 = img1.crop((0, 0, min(img1.size[0], img2.size[0]), min(img1.size[1], img2.size[1])))
|
| 43 |
img2 = img2.crop((0, 0, min(img1.size[0], img2.size[0]), min(img1.size[1], img2.size[1])))
|
|
|
|
|
|
|
| 44 |
img1.save(frame1)
|
| 45 |
img2.save(frame2)
|
| 46 |
|