PaddlePaddle / PaddleSeg

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.
https://arxiv.org/abs/2101.06175
Apache License 2.0
8.65k stars 1.68k forks source link

Unet训练验证时,MIOU为0,precission为0,recall为0,Kappa为0。 #2755

Closed SweepingMonkWWW closed 1 year ago

SweepingMonkWWW commented 1 year ago

当使用unet模型训练验证时,MIOU为0,precission为0,recall为0,Kappa为0。配置文件相同,其他模型均正常,请问是什么原因? f677549e4ca3114859a299743d636ab

SweepingMonkWWW commented 1 year ago

保证数据集一致,配置文件一致,利用其他模型训练时则正常,unet训练就不行。下图为其他模型评估时的结果。 5826fd68be135e5dd799536cbb91c9b

SweepingMonkWWW commented 1 year ago

相关配置文件如下 base: '../base/cityscapes.yml'

batch_size: 2 iters: 1000

model: type: UNetPlusPlus in_channels: 3 num_classes: 2 use_deconv: False align_corners: False pretrained: Null is_ds: True

batch_size: 2 iters: 80000

train_dataset: type: Dataset dataset_root: data/positive_data train_path: data/positive_data/train_list.txt num_classes: 2 mode: train transforms:

val_dataset: type: Dataset dataset_root: data/positive_data val_path: data/positive_data/val_list.txt num_classes: 2 mode: val transforms:

optimizer: type: sgd momentum: 0.9 weight_decay: 4.0e-5

lr_scheduler: type: PolynomialDecay learning_rate: 0.01 end_lr: 0 power: 0.9

loss: types:

wuyefeilin commented 1 year ago

只训练了1000iter,太短了,模型还不能收敛,训练长一些

github-actions[bot] commented 1 year ago

This issue has been automatically marked as stale because it has not had recent activity. It will be closed in 7 days if no further activity occurs. Thank you for your contributions.