zhixuhao / unet

unet for image segmentation
MIT License
4.6k stars 2k forks source link

问题汇总 #256

Open ghLcd9dG opened 7 months ago

ghLcd9dG commented 7 months ago

1、TypeError: ('Keyword argument not understood:', 'input')

A: Model文件里面的input修改为 model = Model(inputs=inputs, outputs=conv10)

2、cannot write mode F as PNG

A: 参见问题3,修改saveResult

3、训练完结果是黑色的/白色的

A: 训练的轮数太少了,loss不下降 / saveResult 函数没有改过来

for i, item in enumerate(npyfile):
        img = (
            labelVisualize(num_class, COLOR_DICT, item)
            if flag_multi_class
            else item[:, :, 0]
        )
        img = (img > 0.5).astype(np.uint8)  # .reshape(256, 256)
        img = img * 255
        cv2.imwrite(os.path.join(save_path, "%d_predict.png" % i), img)

4、结果应该是什么样子的

image

5、修改好的代码/成品

正在提PR,地址:https://github.com/Fab-Liu/unet.git

2462370039 commented 5 months ago

您好,跑您刚修改后的代码还是输出全白图片,loss也是nan 43/300 [===>..........................] - ETA: 46s - loss: nan - accuracy: 0.5420 这是什么原因?