Upload folder using huggingface_hub
Browse files- __pycache__/predict.cpython-311.pyc +0 -0
- predict.py +2 -2
__pycache__/predict.cpython-311.pyc
CHANGED
|
Binary files a/__pycache__/predict.cpython-311.pyc and b/__pycache__/predict.cpython-311.pyc differ
|
|
|
predict.py
CHANGED
|
@@ -240,7 +240,7 @@ def predict_next_frame(ens, context_frames: np.ndarray) -> np.ndarray:
|
|
| 240 |
ens.direct_cache = []
|
| 241 |
for i in range(PRED_FRAMES):
|
| 242 |
frame = np.transpose(predicted_np[i], (1, 2, 0))
|
| 243 |
-
frame = np.round(frame * 255 + 0.
|
| 244 |
ens.direct_cache.append(frame)
|
| 245 |
|
| 246 |
result = ens.direct_cache[ens.cache_step]
|
|
@@ -272,7 +272,7 @@ def predict_next_frame(ens, context_frames: np.ndarray) -> np.ndarray:
|
|
| 272 |
ens.direct_cache = []
|
| 273 |
for i in range(PRED_FRAMES):
|
| 274 |
frame = np.transpose(predicted_np[i], (1, 2, 0))
|
| 275 |
-
frame = np.round(frame * 255
|
| 276 |
ens.direct_cache.append(frame)
|
| 277 |
|
| 278 |
result = ens.direct_cache[ens.cache_step]
|
|
|
|
| 240 |
ens.direct_cache = []
|
| 241 |
for i in range(PRED_FRAMES):
|
| 242 |
frame = np.transpose(predicted_np[i], (1, 2, 0))
|
| 243 |
+
frame = np.round(frame * 255 + 0.2).clip(0, 255).astype(np.uint8)
|
| 244 |
ens.direct_cache.append(frame)
|
| 245 |
|
| 246 |
result = ens.direct_cache[ens.cache_step]
|
|
|
|
| 272 |
ens.direct_cache = []
|
| 273 |
for i in range(PRED_FRAMES):
|
| 274 |
frame = np.transpose(predicted_np[i], (1, 2, 0))
|
| 275 |
+
frame = np.round(frame * 255).clip(0, 255).astype(np.uint8)
|
| 276 |
ens.direct_cache.append(frame)
|
| 277 |
|
| 278 |
result = ens.direct_cache[ens.cache_step]
|