deeplabv3-plus-pytorch
deeplabv3-plus-pytorch copied to clipboard
bubbliiiing
这是一个deeplabv3-plus-pytorch的源码，可以用于训练自己的模型。
模型输出问题
老师好，想请问一下，默认导出onnx之后，模型输出是三通道的RGB图像，怎么将输出改为0或1的标签呀，比如我就对两个类进行分割（背景和目标），然后我希望输出二值图（背景为0，目标为1），就好像训练的数据集的标签一样 模型转onnx格式之后，输出是3通道的float图，仔细观察后通道1和通道2都有点奇怪，通道3是我想要的（需要格式转换），怎么控制网络直接输出灰度标签呢
關於訓練測試的一些問題
1.目前程式我只要將pretrained設定為True會跳出這個錯誤，請問如何解決?(我已經預先下載好.pth檔並且放在model_data)  2.請問要如何在程式新增計算dice值?
我替换为自己的数据集进行训练，但是一直遇到ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 256, 1, 1])，这个报错，百度了都说是batch_size可能多出来了，把drop_last设置为true就好了。可是我看你的代码里是设置的true啊，请问up怎么解决呢，我这边cuda：10.1 pytorch：1.2.0
请问下面这条公式出处是哪里呀？或者是作者您写的吗？ https://github.com/bubbliiiing/deeplabv3-plus-pytorch/blob/916132716464807865d71f6e8234086e38754929/train.py#L413
