CE-Net
CE-Net copied to clipboard
The manuscript has been accepted in TMI.
Hi! Could you please provide Set_A.txt and Set_B.txt for experiment on ORIGA dataset? I searched online for how ORIGA dataset was divided into training set and test set, but found...
RuntimeError: The size of tensor a (38) must match the size of tensor b (37) at non-singleton dimension 2 在运行test_center中d4 = self.decoder4(e4) + e3时,pycharm报错,请问有人知道原因吗?非常感谢
I can see the data augmentation in Readme. But I can‘t find the data.py in your fold.
it seems Unet in your code can outperform CE_Net in my training. I directly used your code and retrained the Unet, I got a accuracy of [acc: 0.956 | sen:...
wrong test
I run test_cenet.py, but i got the all black or all white mask. how can i solve this? and i want to know where is your pretrained model? Thanks @Guzaiwang
i run the main.py, parameter is default,then run the test_cenet.py,but the prediction is not good,could you give me some suggestion?thank you very much! 
request
Can you configure the environment for the next code? For example, torch version? Thank you
预测时出现错误
RuntimeError: Error(s) in loading state_dict for CE_Net_: Missing key(s) in state_dict: "firstconv.weight", "firstbn.weight", "firstbn.bias", "firstbn.running_mean", "firstbn.running_var", "encoder1.0.conv1.weight", "encoder1.0.bn1.weight", "encoder1.0.bn1.bias", "encoder1.0.bn1.running_mean", "encoder1.0.bn1.running_var", "encoder1.0.conv2.weight", "encoder1.0.bn2.weight", "encoder1.0.bn2.bias", "encoder1.0.bn2.running_mean", "encoder1.0.bn2.running_var", "encoder1.1.conv1.weight", "encoder1.1.bn1.weight", "encoder1.1.bn1.bias", "encoder1.1.bn1.running_mean",...
就是你代码里并没有第一层输出网络和后面的连接,但是你论文中的图却有,不知道是不是我没有看懂你的代码。  