nabsabraham / focal-tversky-unet

This repo contains the code for our paper "A novel focal Tversky loss function and improved Attention U-Net for lesion segmentation" accepted at IEEE ISBI 2019.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Unet gating signal typo?

ivanvoid opened this issue · comments

g3 = UnetGatingSignal(up1, is_batchnorm=True, name='g3')

Hi, thanks for publishing code. I think you made a typo in line 268 in file newmodels.py. I sure it must be:
g3 = UnetGatingSignal(up2, is_batchnorm=True, name='g3'), can you conform it? or explain why do you use signal in third attention block from up1 with conv2?
Thanks~