can i use new arch? - Githubissues

mjkwon2021 / CAT-Net

Official code for CAT-Net: Compression Artifact Tracing Network. Image manipulation detection and localization.

210 stars 25 forks source link

can i use new arch? #47

Open shravanti1 opened 6 months ago

shravanti1 commented 6 months ago

Hi I want to change HRNet to UNET. What changes I should do? I have changed HRNet architecture to UNET but for pretraining I am unable to find UNET weights with ImageNet available. Pls Help

CauchyComplete commented 6 months ago

Hi, Since you introduced the new architecture, you should pretrain it yourself with ImageNet 😄