search

pmh9960 / iColoriT

Official PyTorch implementation of "iColoriT: Towards Propagating Local Hint to the Right Region in Interactive Colorization by Leveraging Vision Transformer." (WACV 2023)
MIT License
69 stars 14 forks source link

Stabilization Layer #15

Open Marcelo5444 opened 11 months ago

Marcelo5444 commented 11 months ago

The stabilization layer were it can be found? is the here: https://github.com/pmh9960/iColoriT/blob/8a02415aea6f6cc144650b47f38e58476f4e2a0f/iColoriT_demo/modeling.py#L244

YeolJ00 commented 11 months ago

Yes. That is the stabilization layer.

Marcelo5444 commented 11 months ago

Hi, checking the stabilizing layer, in our paper diagram its that features forwarded across the Conv2d, keeep the same dimension however, when running the code I see that the: nn.Conv2d(192, 512, kernel_size=3, stride=1, padding=1, padding_mode='reflect')

Could you explain that, please