XPixelGroup / HAT

CVPR2023 - Activating More Pixels in Image Super-Resolution Transformer Arxiv - HAT: Hybrid Attention Transformer for Image Restoration
Apache License 2.0
1.14k stars 134 forks source link

How to modify basicsr code and apply to HAT model? #123

Open uigeunahn opened 6 months ago

uigeunahn commented 6 months ago

Firstly, thanks for your great work. I really want to apply the model to my project. My project requires input of 16bit image which is Xray dcm. Before when I was using BSRN which is also SR Model based on basicsr, I could modify the basicsr/utils/image_util.py to read and save as Xray dcm when training and testing. But in this case, unlike BSRN there is no basicsr folder in the repository. How can I also apply the change the basicsr code? image

Anyone who knows the answer, please let me know. Really want to try this SOTA model on medical image super resolution. Thanks

CLL112 commented 3 months ago

This project uses the installation package form of BasicSR, so there are no files like 'basicsr/utils/image_util.py.' You need to manually copy the code you want to modify

dtsiap commented 1 month ago

@uigeunahn Were you able to solve it? @CLL112 Which would the target directory be in that case? Where should the BasicSR files be pasted to be modified? Because HAT is accessing it as a toolbox.