Line patterns remain where mosaic blocks edges were

[Milincho]

Using the clean_youknow_video.pth model to remove typical jav censorship mosaics I get pretty godd results, but unfortunately there are always a recognizable pattern were the edges of blocks were in the original mosaic:

https://i.imgur.com/6IYWcun.png

Is there any way to fix this?

[HypoX64]

This may caused by the last shortcut. I will try to replace it.

  def forward(self, stream, previous):
      this_shortcut = stream[:,:,self.N]
      stream = self.encoder3d(stream)
      stream = stream.reshape(stream.size(0),stream.size(1),stream.size(3),stream.size(4))
      previous = self.encoder2d(previous)
      x = stream + previous
      x = self.blocks(x)
      x = self.decoder(x)
      x = x+this_shortcut
      x = self.limiter(x)
      return x

[Milincho]

Great, thank you! any estimate when a new binary version will be released? :-)