Megvii-BaseDetection / TreeFilter-Torch

Learnable Tree Filter for Structure-preserving Feature Transform
MIT License
139 stars 13 forks source link

have you try hrnet +ocr + treefilter? #7

Open AncientRemember opened 3 years ago

StevenGrove commented 3 years ago

I have not tried this specific case, but I think it will work fine. It is easy to implement in your codebase. You can add a tree filter after each "upsample" or "downsample" operations in the HRNet. Please refer to decoder usage and encoder usage for more details.