search

wutianyiRosun / CGNet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]
MIT License
262 stars 51 forks source link

classweights #9

Closed yyfyan closed 5 years ago

yyfyan commented 5 years ago

I find the code used classweights is defferent from OCNet. https://github.com/PkuRainBow/OCNet.pytorch/blob/master/utils/loss.py

yyfyan commented 5 years ago

the code need changed: hist = np.histogram(label_img, self.classes,range=(0,18))

wutianyiRosun commented 5 years ago

@yyfyan thank u.