请问如何修改做multiclass - Githubissues

JavisPeng / u_net_liver

MIT License

201 stars 75 forks source link

请问如何修改做multiclass #2

Closed lfb-1 closed 5 years ago

lfb-1 commented 5 years ago

你好，请问如何修改让他可以训练多个class，需要修改哪些部分？谢谢

JavisPeng commented 5 years ago

模型最后层的通道数为类别数，去掉sigmoid层，loss函数使用CrossEntropyLoss