use CU-net for denoising

zhiqiangdon / CU-Net

Code for "Quantized Densely Connected U-Nets for Efficient Landmark Localization" (ECCV 2018) and "CU-Net: Coupled U-Nets" (BMVC 2018 oral)

Apache License 2.0

227 stars 40 forks source link

use CU-net for denoising #14

Open HosRahimi opened 2 months ago

HosRahimi commented 2 months ago

hi im wnt to use CU-net for denoising but when in fed an image with shape (1, 3, 128, 128), the output shape of 'create_cu_net' function is (1, 3. 32, 32). how can i fix it? i have to use 'TransitionUp' module? or what?