wutianyiRosun / CGNet

CGNet: A Light-weight Context Guided Network for Semantic Segmentation [IEEE Transactions on Image Processing 2020]

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

classweights

yyfyan opened this issue · comments

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

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

commented

@yyfyan thank u.