U-Net模型eval和vis问题
Created by: gxqnba
使用的是自己的数据集(gt图像是灰度图像,背景黑色,目标为白色),通过了数据检校 训练好之后进行eval:python pdseg/eval.py --use_gpu --cfg ./configs/unet_optic.yaml,报错如下: [EVAL]step=1 loss=0.00022 acc=1.0000 IoU=0.5000 step/sec=2.59 | ETA 00:00:01 [EVAL]step=2 loss=0.00023 acc=1.0000 IoU=0.5000 step/sec=8.15 | ETA 00:00:00 [EVAL]step=3 loss=0.00025 acc=1.0000 IoU=0.5000 step/sec=8.96 | ETA 00:00:00 [EVAL]step=4 loss=0.00024 acc=1.0000 IoU=0.5000 step/sec=9.09 | ETA 00:00:00 [EVAL]step=5 loss=0.00024 acc=1.0000 IoU=0.5000 step/sec=16.76 | ETA 00:00:00 [EVAL]#image=18 acc=1.0000 IoU=0.5000 [EVAL]Category IoU: [1.0000 0.0000] [EVAL]Category Acc: [1.0000 0.0000] /home/aistudio/work/PaddleSeg/pdseg/metrics.py:144: RuntimeWarning: invalid value encountered in double_scalars kappa = (po - pe) / (1 - pe) [EVAL]Kappa:nan
另外,在进行vis时:python pdseg/vis.py --use_gpu --cfg ./configs/unet_optic.yaml,得到的分割结果是一张全黑图像