bubbliiiing / unet-pytorch

这是一个unet-pytorch的源码,可以训练自己的模型

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

dice_loss, ce_loss, focal_loss的输入和traget的区别

BaronDuan opened this issue · comments

你好:
我在看博主的代码的时候发现,发现代码中focal_loss和CE_loss的输入是jpg, target是png,但是dice_loss的输入是jpg,target却是labels,这是为什么呀?为什么博主大大要用不同的真值呀?

image
详见dataloader.py和unet_training.py