-
Notifications
You must be signed in to change notification settings - Fork 76
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
预测时出现错误 #21
Comments
use strict = False mode when loading the pretrained model. it works ! |
I can't find the pretrained model (CE-Net_DRIVE_auc_0.9819.th), where do you get it??? |
You should train on the dataset.
…---原始邮件---
发件人: "lzw3232"<[email protected]>
发送时间: 2020年10月22日(周四) 下午5:36
收件人: "Guzaiwang/CE-Net"<[email protected]>;
抄送: "ShenghaiLiao"<[email protected]>;"Comment"<[email protected]>;
主题: Re: [Guzaiwang/CE-Net] 预测时出现错误 (#21)
I can't find the pretrained model (CE-Net_DRIVE_auc_0.9819.th), where do you get it???
—
You are receiving this because you commented.
Reply to this email directly, view it on GitHub, or unsubscribe.
|
|
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", "encoder1.1.bn1.running_var", "encoder1.1.conv2.weight", "encoder1.1.bn2.weight", "encoder1.1.bn2.bias", "encoder1.1.bn2.running_mean", "encoder1.1.bn2.running_var", "encoder1.2.conv1.weight", "encoder1.2.bn1.weight", "encoder1.2.bn1.bias", "encoder1.2.bn1.running_mean", "encoder1.2.bn1.running_var", "encoder1.2.conv2.weight", "encoder1.2.bn2.weight", "encoder1.2.bn2.bias", "encoder1.2.bn2.running_mean", "encoder1.2.bn2.running_var", "encoder2.0.conv1.weight", "encoder2.0.bn1.weight", "encoder2.0.bn1.bias", "encoder2.0.bn1.running_mean", "encoder2.0.bn1.running_var", "encoder2.0.conv2.weight", "encoder2.0.bn2.weight", "encoder2.0.bn2.bias", "encoder2.0.bn2.running_mean", "encoder2.0.bn2.running_var", "encoder2.0.downsample.0.weight", "encoder2.0.downsample.1.weight", "encoder2.0.downsample.1.bias", "encoder2.0.downsample.1.running_mean", "encoder2.0.downsample.1.running_var", "encoder2.1.conv1.weight", "encoder2.1.bn1.weight", "encoder2.1.bn1.bias", "encoder2.1.bn1.running_mean", "encoder2.1.bn1.running_var", "encoder2.1.conv2.weight", "encoder2.1.bn2.weight", "encoder2.1.bn2.bias", "encoder2.1.bn2.running_mean", "encoder2.1.bn2.running_var", "encoder2.2.conv1.weight", "encoder2.2.bn1.weight", "encoder2.2.bn1.bias", "encoder2.2.bn1.running_mean", "encoder2.2.bn1.running_var", "encoder2.2.conv2.weight", "encoder2.2.bn2.weight", "encoder2.2.bn2.bias", "encoder2.2.bn2.running_mean", "encoder2.2.bn2.running_var", "encoder2.3.conv1.weight", "encoder2.3.bn1.weight", "encoder2.3.bn1.bias", "encoder2.3.bn1.running_mean", "encoder2.3.bn1.running_var", "encoder2.3.conv2.weight", "encoder2.3.bn2.weight", "encoder2.3.bn2.bias", "encoder2.3.bn2.running_mean", "encoder2.3.bn2.running_var", "encoder3.0.conv1.weight", "encoder3.0.bn1.weight", "encoder3.0.bn1.bias", "encoder3.0.bn1.running_mean", "encoder3.0.bn1.running_var", "encoder3.0.conv2.weight", "encoder3.0.bn2.weight", "encoder3.0.bn2.bias", "encoder3.0.bn2.running_mean", "encoder3.0.bn2.running_var", "encoder3.0.downsample.0.weight", "encoder3.0.downsample.1.weight", "encoder3.0.downsample.1.bias", "encoder3.0.downsample.1.running_mean", "encoder3.0.downsample.1.running_var", "encoder3.1.conv1.weight", "encoder3.1.bn1.weight", "encoder3.1.bn1.bias", "encoder3.1.bn1.running_mean", "encoder3.1.bn1.running_var", "encoder3.1.conv2.weight", "encoder3.1.bn2.weight", "encoder3.1.bn2.bias", "encoder3.1.bn2.running_mean", "encoder3.1.bn2.running_var", "encoder3.2.conv1.weight", "encoder3.2.bn1.weight", "encoder3.2.bn1.bias", "encoder3.2.bn1.running_mean", "encoder3.2.bn1.running_var", "encoder3.2.conv2.weight", "encoder3.2.bn2.weight", "encoder3.2.bn2.bias", "encoder3.2.bn2.running_mean", "encoder3.2.bn2.running_var", "encoder3.3.conv1.weight", "encoder3.3.bn1.weight", "encoder3.3.bn1.bias", "encoder3.3.bn1.running_mean", "encoder3.3.bn1.running_var", "encoder3.3.conv2.weight", "encoder3.3.bn2.weight", "encoder3.3.bn2.bias", "encoder3.3.bn2.running_mean", "encoder3.3.bn2.running_var", "encoder3.4.conv1.weight", "encoder3.4.bn1.weight", "encoder3.4.bn1.bias", "encoder3.4.bn1.running_mean", "encoder3.4.bn1.running_var", "encoder3.4.conv2.weight", "encoder3.4.bn2.weight", "encoder3.4.bn2.bias", "encoder3.4.bn2.running_mean", "encoder3.4.bn2.running_var", "encoder3.5.conv1.weight", "encoder3.5.bn1.weight", "encoder3.5.bn1.bias", "encoder3.5.bn1.running_mean", "encoder3.5.bn1.running_var", "encoder3.5.conv2.weight", "encoder3.5.bn2.weight", "encoder3.5.bn2.bias", "encoder3.5.bn2.running_mean", "encoder3.5.bn2.running_var", "encoder4.0.conv1.weight", "encoder4.0.bn1.weight", "encoder4.0.bn1.bias", "encoder4.0.bn1.running_mean", "encoder4.0.bn1.running_var", "encoder4.0.conv2.weight", "encoder4.0.bn2.weight", "encoder4.0.bn2.bias", "encoder4.0.bn2.running_mean", "encoder4.0.bn2.running_var", "encoder4.0.downsample.0.weight", "encoder4.0.downsample.1.weight", "encoder4.0.downsample.1.bias", "encoder4.0.downsample.1.running_mean", "encoder4.0.downsample.1.running_var", "encoder4.1.conv1.weight", "encoder4.1.bn1.weight", "encoder4.1.bn1.bias", "encoder4.1.bn1.running_mean", "encoder4.1.bn1.running_var", "encoder4.1.conv2.weight", "encoder4.1.bn2.weight", "encoder4.1.bn2.bias", "encoder4.1.bn2.running_mean", "encoder4.1.bn2.running_var", "encoder4.2.conv1.weight", "encoder4.2.bn1.weight", "encoder4.2.bn1.bias", "encoder4.2.bn1.running_mean", "encoder4.2.bn1.running_var", "encoder4.2.conv2.weight", "encoder4.2.bn2.weight", "encoder4.2.bn2.bias", "encoder4.2.bn2.running_mean", "encoder4.2.bn2.running_var", "dblock.dilate1.weight", "dblock.dilate1.bias", "dblock.dilate2.weight", "dblock.dilate2.bias", "dblock.dilate3.weight", "dblock.dilate3.bias", "dblock.conv1x1.weight", "dblock.conv1x1.bias", "spp.conv.weight", "spp.conv.bias", "decoder4.conv1.weight", "decoder4.conv1.bias", "decoder4.norm1.weight", "decoder4.norm1.bias", "decoder4.norm1.running_mean", "decoder4.norm1.running_var", "decoder4.deconv2.weight", "decoder4.deconv2.bias", "decoder4.norm2.weight", "decoder4.norm2.bias", "decoder4.norm2.running_mean", "decoder4.norm2.running_var", "decoder4.conv3.weight", "decoder4.conv3.bias", "decoder4.norm3.weight", "decoder4.norm3.bias", "decoder4.norm3.running_mean", "decoder4.norm3.running_var", "decoder3.conv1.weight", "decoder3.conv1.bias", "decoder3.norm1.weight", "decoder3.norm1.bias", "decoder3.norm1.running_mean", "decoder3.norm1.running_var", "decoder3.deconv2.weight", "decoder3.deconv2.bias", "decoder3.norm2.weight", "decoder3.norm2.bias", "decoder3.norm2.running_mean", "decoder3.norm2.running_var", "decoder3.conv3.weight", "decoder3.conv3.bias", "decoder3.norm3.weight", "decoder3.norm3.bias", "decoder3.norm3.running_mean", "decoder3.norm3.running_var", "decoder2.conv1.weight", "decoder2.conv1.bias", "decoder2.norm1.weight", "decoder2.norm1.bias", "decoder2.norm1.running_mean", "decoder2.norm1.running_var", "decoder2.deconv2.weight", "decoder2.deconv2.bias", "decoder2.norm2.weight", "decoder2.norm2.bias", "decoder2.norm2.running_mean", "decoder2.norm2.running_var", "decoder2.conv3.weight", "decoder2.conv3.bias", "decoder2.norm3.weight", "decoder2.norm3.bias", "decoder2.norm3.running_mean", "decoder2.norm3.running_var", "decoder1.conv1.weight", "decoder1.conv1.bias", "decoder1.norm1.weight", "decoder1.norm1.bias", "decoder1.norm1.running_mean", "decoder1.norm1.running_var", "decoder1.deconv2.weight", "decoder1.deconv2.bias", "decoder1.norm2.weight", "decoder1.norm2.bias", "decoder1.norm2.running_mean", "decoder1.norm2.running_var", "decoder1.conv3.weight", "decoder1.conv3.bias", "decoder1.norm3.weight", "decoder1.norm3.bias", "decoder1.norm3.running_mean", "decoder1.norm3.running_var", "finaldeconv1.weight", "finaldeconv1.bias", "finalconv2.weight", "finalconv2.bias", "finalconv3.weight", "finalconv3.bias".
Unexpected key(s) in state_dict: "module.firstconv.weight", "module.firstbn.weight", "module.firstbn.bias", "module.firstbn.running_mean", "module.firstbn.running_var", "module.firstbn.num_batches_tracked", "module.encoder1.0.conv1.weight", "module.encoder1.0.bn1.weight", "module.encoder1.0.bn1.bias", "module.encoder1.0.bn1.running_mean", "module.encoder1.0.bn1.running_var", "module.encoder1.0.bn1.num_batches_tracked", "module.encoder1.0.conv2.weight", "module.encoder1.0.bn2.weight", "module.encoder1.0.bn2.bias", "module.encoder1.0.bn2.running_mean", "module.encoder1.0.bn2.running_var", "module.encoder1.0.bn2.num_batches_tracked", "module.encoder1.1.conv1.weight", "module.encoder1.1.bn1.weight", "module.encoder1.1.bn1.bias", "module.encoder1.1.bn1.running_mean", "module.encoder1.1.bn1.running_var", "module.encoder1.1.bn1.num_batches_tracked", "module.encoder1.1.conv2.weight", "module.encoder1.1.bn2.weight", "module.encoder1.1.bn2.bias", "module.encoder1.1.bn2.running_mean", "module.encoder1.1.bn2.running_var", "module.encoder1.1.bn2.num_batches_tracked", "module.encoder1.2.conv1.weight", "module.encoder1.2.bn1.weight", "module.encoder1.2.bn1.bias", "module.encoder1.2.bn1.running_mean", "module.encoder1.2.bn1.running_var", "module.encoder1.2.bn1.num_batches_tracked", "module.encoder1.2.conv2.weight", "module.encoder1.2.bn2.weight", "module.encoder1.2.bn2.bias", "module.encoder1.2.bn2.running_mean", "module.encoder1.2.bn2.running_var", "module.encoder1.2.bn2.num_batches_tracked", "module.encoder2.0.conv1.weight", "module.encoder2.0.bn1.weight", "module.encoder2.0.bn1.bias", "module.encoder2.0.bn1.running_mean", "module.encoder2.0.bn1.running_var", "module.encoder2.0.bn1.num_batches_tracked", "module.encoder2.0.conv2.weight", "module.encoder2.0.bn2.weight", "module.encoder2.0.bn2.bias", "module.encoder2.0.bn2.running_mean", "module.encoder2.0.bn2.running_var", "module.encoder2.0.bn2.num_batches_tracked", "module.encoder2.0.downsample.0.weight", "module.encoder2.0.downsample.1.weight", "module.encoder2.0.downsample.1.bias", "module.encoder2.0.downsample.1.running_mean", "module.encoder2.0.downsample.1.running_var", "module.encoder2.0.downsample.1.num_batches_tracked", "module.encoder2.1.conv1.weight", "module.encoder2.1.bn1.weight", "module.encoder2.1.bn1.bias", "module.encoder2.1.bn1.running_mean", "module.encoder2.1.bn1.running_var", "module.encoder2.1.bn1.num_batches_tracked", "module.encoder2.1.conv2.weight", "module.encoder2.1.bn2.weight", "module.encoder2.1.bn2.bias", "module.encoder2.1.bn2.running_mean", "module.encoder2.1.bn2.running_var", "module.encoder2.1.bn2.num_batches_tracked", "module.encoder2.2.conv1.weight", "module.encoder2.2.bn1.weight", "module.encoder2.2.bn1.bias", "module.encoder2.2.bn1.running_mean", "module.encoder2.2.bn1.running_var", "module.encoder2.2.bn1.num_batches_tracked", "module.encoder2.2.conv2.weight", "module.encoder2.2.bn2.weight", "module.encoder2.2.bn2.bias", "module.encoder2.2.bn2.running_mean", "module.encoder2.2.bn2.running_var", "module.encoder2.2.bn2.num_batches_tracked", "module.encoder2.3.conv1.weight", "module.encoder2.3.bn1.weight", "module.encoder2.3.bn1.bias", "module.encoder2.3.bn1.running_mean", "module.encoder2.3.bn1.running_var", "module.encoder2.3.bn1.num_batches_tracked", "module.encoder2.3.conv2.weight", "module.encoder2.3.bn2.weight", "module.encoder2.3.bn2.bias", "module.encoder2.3.bn2.running_mean", "module.encoder2.3.bn2.running_var", "module.encoder2.3.bn2.num_batches_tracked", "module.encoder3.0.conv1.weight", "module.encoder3.0.bn1.weight", "module.encoder3.0.bn1.bias", "module.encoder3.0.bn1.running_mean", "module.encoder3.0.bn1.running_var", "module.encoder3.0.bn1.num_batches_tracked", "module.encoder3.0.conv2.weight", "module.encoder3.0.bn2.weight", "module.encoder3.0.bn2.bias", "module.encoder3.0.bn2.running_mean", "module.encoder3.0.bn2.running_var", "module.encoder3.0.bn2.num_batches_tracked", "module.encoder3.0.downsample.0.weight", "module.encoder3.0.downsample.1.weight", "module.encoder3.0.downsample.1.bias", "module.encoder3.0.downsample.1.running_mean", "module.encoder3.0.downsample.1.running_var", "module.encoder3.0.downsample.1.num_batches_tracked", "module.encoder3.1.conv1.weight", "module.encoder3.1.bn1.weight", "module.encoder3.1.bn1.bias", "module.encoder3.1.bn1.running_mean", "module.encoder3.1.bn1.running_var", "module.encoder3.1.bn1.num_batches_tracked", "module.encoder3.1.conv2.weight", "module.encoder3.1.bn2.weight", "module.encoder3.1.bn2.bias", "module.encoder3.1.bn2.running_mean", "module.encoder3.1.bn2.running_var", "module.encoder3.1.bn2.num_batches_tracked", "module.encoder3.2.conv1.weight", "module.encoder3.2.bn1.weight", "module.encoder3.2.bn1.bias", "module.encoder3.2.bn1.running_mean", "module.encoder3.2.bn1.running_var", "module.encoder3.2.bn1.num_batches_tracked", "module.encoder3.2.conv2.weight", "module.encoder3.2.bn2.weight", "module.encoder3.2.bn2.bias", "module.encoder3.2.bn2.running_mean", "module.encoder3.2.bn2.running_var", "module.encoder3.2.bn2.num_batches_tracked", "module.encoder3.3.conv1.weight", "module.encoder3.3.bn1.weight", "module.encoder3.3.bn1.bias", "module.encoder3.3.bn1.running_mean", "module.encoder3.3.bn1.running_var", "module.encoder3.3.bn1.num_batches_tracked", "module.encoder3.3.conv2.weight", "module.encoder3.3.bn2.weight", "module.encoder3.3.bn2.bias", "module.encoder3.3.bn2.running_mean", "module.encoder3.3.bn2.running_var", "module.encoder3.3.bn2.num_batches_tracked", "module.encoder3.4.conv1.weight", "module.encoder3.4.bn1.weight", "module.encoder3.4.bn1.bias", "module.encoder3.4.bn1.running_mean", "module.encoder3.4.bn1.running_var", "module.encoder3.4.bn1.num_batches_tracked", "module.encoder3.4.conv2.weight", "module.encoder3.4.bn2.weight", "module.encoder3.4.bn2.bias", "module.encoder3.4.bn2.running_mean", "module.encoder3.4.bn2.running_var", "module.encoder3.4.bn2.num_batches_tracked", "module.encoder3.5.conv1.weight", "module.encoder3.5.bn1.weight", "module.encoder3.5.bn1.bias", "module.encoder3.5.bn1.running_mean", "module.encoder3.5.bn1.running_var", "module.encoder3.5.bn1.num_batches_tracked", "module.encoder3.5.conv2.weight", "module.encoder3.5.bn2.weight", "module.encoder3.5.bn2.bias", "module.encoder3.5.bn2.running_mean", "module.encoder3.5.bn2.running_var", "module.encoder3.5.bn2.num_batches_tracked", "module.encoder4.0.conv1.weight", "module.encoder4.0.bn1.weight", "module.encoder4.0.bn1.bias", "module.encoder4.0.bn1.running_mean", "module.encoder4.0.bn1.running_var", "module.encoder4.0.bn1.num_batches_tracked", "module.encoder4.0.conv2.weight", "module.encoder4.0.bn2.weight", "module.encoder4.0.bn2.bias", "module.encoder4.0.bn2.running_mean", "module.encoder4.0.bn2.running_var", "module.encoder4.0.bn2.num_batches_tracked", "module.encoder4.0.downsample.0.weight", "module.encoder4.0.downsample.1.weight", "module.encoder4.0.downsample.1.bias", "module.encoder4.0.downsample.1.running_mean", "module.encoder4.0.downsample.1.running_var", "module.encoder4.0.downsample.1.num_batches_tracked", "module.encoder4.1.conv1.weight", "module.encoder4.1.bn1.weight", "module.encoder4.1.bn1.bias", "module.encoder4.1.bn1.running_mean", "module.encoder4.1.bn1.running_var", "module.encoder4.1.bn1.num_batches_tracked", "module.encoder4.1.conv2.weight", "module.encoder4.1.bn2.weight", "module.encoder4.1.bn2.bias", "module.encoder4.1.bn2.running_mean", "module.encoder4.1.bn2.running_var", "module.encoder4.1.bn2.num_batches_tracked", "module.encoder4.2.conv1.weight", "module.encoder4.2.bn1.weight", "module.encoder4.2.bn1.bias", "module.encoder4.2.bn1.running_mean", "module.encoder4.2.bn1.running_var", "module.encoder4.2.bn1.num_batches_tracked", "module.encoder4.2.conv2.weight", "module.encoder4.2.bn2.weight", "module.encoder4.2.bn2.bias", "module.encoder4.2.bn2.running_mean", "module.encoder4.2.bn2.running_var", "module.encoder4.2.bn2.num_batches_tracked", "module.dblock.dilate1.weight", "module.dblock.dilate1.bias", "module.dblock.dilate2.weight", "module.dblock.dilate2.bias", "module.dblock.dilate3.weight", "module.dblock.dilate3.bias", "module.dblock.conv1x1.weight", "module.dblock.conv1x1.bias", "module.spp.conv.weight", "module.spp.conv.bias", "module.decoder4.conv1.weight", "module.decoder4.conv1.bias", "module.decoder4.norm1.weight", "module.decoder4.norm1.bias", "module.decoder4.norm1.running_mean", "module.decoder4.norm1.running_var", "module.decoder4.norm1.num_batches_tracked", "module.decoder4.deconv2.weight", "module.decoder4.deconv2.bias", "module.decoder4.norm2.weight", "module.decoder4.norm2.bias", "module.decoder4.norm2.running_mean", "module.decoder4.norm2.running_var", "module.decoder4.norm2.num_batches_tracked", "module.decoder4.conv3.weight", "module.decoder4.conv3.bias", "module.decoder4.norm3.weight", "module.decoder4.norm3.bias", "module.decoder4.norm3.running_mean", "module.decoder4.norm3.running_var", "module.decoder4.norm3.num_batches_tracked", "module.decoder3.conv1.weight", "module.decoder3.conv1.bias", "module.decoder3.norm1.weight", "module.decoder3.norm1.bias", "module.decoder3.norm1.running_mean", "module.decoder3.norm1.running_var", "module.decoder3.norm1.num_batches_tracked", "module.decoder3.deconv2.weight", "module.decoder3.deconv2.bias", "module.decoder3.norm2.weight", "module.decoder3.norm2.bias", "module.decoder3.norm2.running_mean", "module.decoder3.norm2.running_var", "module.decoder3.norm2.num_batches_tracked", "module.decoder3.conv3.weight", "module.decoder3.conv3.bias", "module.decoder3.norm3.weight", "module.decoder3.norm3.bias", "module.decoder3.norm3.running_mean", "module.decoder3.norm3.running_var", "module.decoder3.norm3.num_batches_tracked", "module.decoder2.conv1.weight", "module.decoder2.conv1.bias", "module.decoder2.norm1.weight", "module.decoder2.norm1.bias", "module.decoder2.norm1.running_mean", "module.decoder2.norm1.running_var", "module.decoder2.norm1.num_batches_tracked", "module.decoder2.deconv2.weight", "module.decoder2.deconv2.bias", "module.decoder2.norm2.weight", "module.decoder2.norm2.bias", "module.decoder2.norm2.running_mean", "module.decoder2.norm2.running_var", "module.decoder2.norm2.num_batches_tracked", "module.decoder2.conv3.weight", "module.decoder2.conv3.bias", "module.decoder2.norm3.weight", "module.decoder2.norm3.bias", "module.decoder2.norm3.running_mean", "module.decoder2.norm3.running_var", "module.decoder2.norm3.num_batches_tracked", "module.decoder1.conv1.weight", "module.decoder1.conv1.bias", "module.decoder1.norm1.weight", "module.decoder1.norm1.bias", "module.decoder1.norm1.running_mean", "module.decoder1.norm1.running_var", "module.decoder1.norm1.num_batches_tracked", "module.decoder1.deconv2.weight", "module.decoder1.deconv2.bias", "module.decoder1.norm2.weight", "module.decoder1.norm2.bias", "module.decoder1.norm2.running_mean", "module.decoder1.norm2.running_var", "module.decoder1.norm2.num_batches_tracked", "module.decoder1.conv3.weight", "module.decoder1.conv3.bias", "module.decoder1.norm3.weight", "module.decoder1.norm3.bias", "module.decoder1.norm3.running_mean", "module.decoder1.norm3.running_var", "module.decoder1.norm3.num_batches_tracked", "module.finaldeconv1.weight", "module.finaldeconv1.bias", "module.finalconv2.weight", "module.finalconv2.bias", "module.finalconv3.weight", "module.finalconv3.bias".
The text was updated successfully, but these errors were encountered: