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

No PixelShuffle used #14

Closed Marcelo5444 closed 11 months ago

Marcelo5444 commented 11 months ago

Hi, were do you use the pixelshuffle operator? Cant find it

YeolJ00 commented 11 months ago

https://github.com/pmh9960/iColoriT/blob/8a02415aea6f6cc144650b47f38e58476f4e2a0f/engine.py#L66

This is the pixelshuffle operator we use in our paper.