Spaces:

Koi953215
/

DiffIR2VR

Running on Zero

jimmycv07 commited on Jun 28

Commit

661d490

•

1 Parent(s): 2ebea36

uploaded videos from users

Files changed (3) hide show

utils/__pycache__/batch_inference.cpython-310.pyc CHANGED Viewed

Binary files a/utils/__pycache__/batch_inference.cpython-310.pyc and b/utils/__pycache__/batch_inference.cpython-310.pyc differ

utils/__pycache__/helpers.cpython-310.pyc CHANGED Viewed

Binary files a/utils/__pycache__/helpers.cpython-310.pyc and b/utils/__pycache__/helpers.cpython-310.pyc differ

utils/batch_inference.py CHANGED Viewed

@@ -177,7 +177,7 @@ class InferenceLoop:
                     ### set context for saving results
                     self.loop_ctx["file_stem"].append(os.path.splitext(os.path.basename(file_path))[0])
                 # import ipdb; ipdb.set_trace()
-                self.args.final_size = (lqs[0].shape[0] * self.args.upscale, lqs[0].shape[1] * self.args.upscale)
                 for i in range(self.args.n_samples):
                     self.loop_ctx["repeat_idx"] = i
                     yield np.array(lqs)

                     ### set context for saving results
                     self.loop_ctx["file_stem"].append(os.path.splitext(os.path.basename(file_path))[0])
                 # import ipdb; ipdb.set_trace()
+                self.args.final_size = (int(lqs[0].shape[0] * self.args.upscale), int(lqs[0].shape[1] * self.args.upscale))
                 for i in range(self.args.n_samples):
                     self.loop_ctx["repeat_idx"] = i
                     yield np.array(lqs)