这是一个unet-pytorch的源码,可以训练自己的模型
Geek Repo:Geek Repo
Github PK Tool:Github PK Tool
BaronDuan opened this issue 6 months ago · comments
你好: 我在看博主的代码的时候发现,发现代码中focal_loss和CE_loss的输入是jpg, target是png,但是dice_loss的输入是jpg,target却是labels,这是为什么呀?为什么博主大大要用不同的真值呀?
详见dataloader.py和unet_training.py