fix pipeline

QtacierP · Mar 9, 2023 · cbec62d · cbec62d
1 parent d492d08
commit cbec62d
Show file tree

Hide file tree

Showing 11 changed files with 165 additions and 40 deletions.
diff --git a/.gitignore b/.gitignore
@@ -1,4 +1,4 @@
-*/__pycache__
+__pycache__
 pretrained_weights/isecret.pt
 pretrained_weights/led.bin
 pretrained_weights/pcenet.pth

diff --git a/example.ipynb b/example.ipynb
diff --git a/led/backends/arcnet/backend.py b/led/backends/arcnet/backend.py
@@ -41,7 +41,7 @@ def _preprocess(self, input, **kwargs):
         # resize input to defalue size
         if self.hyperparameters.image_size != input.shape[2]:
             input = torch.nn.functional.interpolate(input, 
-            size=self.hyperparameters.image_size, mode='bilinear', align_corners=True)
+            size=self.hyperparameters.image_size, mode='bilinear')
         # unnorm   
         image = input * self.hyperparameters.image_std + self.hyperparameters.image_mean
         image = image.permute(0, 2, 3, 1)

diff --git a/led/backends/isecret/__pycache__/backend.cpython-37.pyc b/led/backends/isecret/__pycache__/backend.cpython-37.pyc
diff --git a/led/backends/isecret/__pycache__/network.cpython-37.pyc b/led/backends/isecret/__pycache__/network.cpython-37.pyc
diff --git a/led/backends/pcenet/__pycache__/backend.cpython-37.pyc b/led/backends/pcenet/__pycache__/backend.cpython-37.pyc
diff --git a/led/backends/pcenet/__pycache__/network.cpython-37.pyc b/led/backends/pcenet/__pycache__/network.cpython-37.pyc
diff --git a/led/backends/pcenet/backend.py b/led/backends/pcenet/backend.py
@@ -37,7 +37,7 @@ def _preprocess(self, input, **kwargs):
         # resize input to defalue size
         if self.hyperparameters.image_size != input.shape[2]:
             input = torch.nn.functional.interpolate(input, 
-            size=self.hyperparameters.image_size, mode='bilinear', align_corners=True)
+            size=self.hyperparameters.image_size, mode='bilinear')
         # unnorm   
         image = input * self.hyperparameters.image_std + self.hyperparameters.image_mean
         image = image.permute(0, 2, 3, 1)

diff --git a/led/backends/scrnet/backend.py b/led/backends/scrnet/backend.py
@@ -67,7 +67,7 @@ def _forward(self, input, mask):
 
     def _post_precess(self, image):
         if self.image_size != image.shape[2]:
-            image = torch.nn.functional.interpolate(image, size=self.image_size, mode='bilinear', align_corners=True)
+            image = torch.nn.functional.interpolate(image, size=self.image_size, mode='bilinear')
         return image
 
     def __call__(self, input):

diff --git a/led/pipelines/__pycache__/led_pipeline.cpython-37.pyc b/led/pipelines/__pycache__/led_pipeline.cpython-37.pyc
diff --git a/led/pipelines/led_pipeline.py b/led/pipelines/led_pipeline.py
@@ -237,8 +237,7 @@ def __call__(
         cond_image = self._pre_process(cond_image).to(self.unet.device)
         if self.backend is not None:
             cond_image = self.backend(cond_image)
-        image = cond_image
-        '''image = randn_tensor(cond_image.shape, generator=generator, device=self.unet.device, dtype=cond_image.dtype)
+        image = randn_tensor(cond_image.shape, generator=generator, device=self.unet.device, dtype=cond_image.dtype)
         # set step values
         self.scheduler.set_timesteps(num_inference_steps)
         max_T = self.num_cond_steps
@@ -255,7 +254,7 @@ def __call__(
             # do x_t -> x_t-1
             image = self.scheduler.step(
                 model_output, t, image, eta=eta, use_clipped_model_output=use_clipped_model_output, generator=generator
-            ).prev_sample'''
+            ).prev_sample
         image = image.cpu().permute(0, 2, 3, 1).numpy()
         if output_type == "pil":
             return self.numpy_to_pil(image)